import com.intellij.testFramework.LightVirtualFile;
-import junit.framework.TestCase;
import org.jetbrains.plugins.ipnb.format.IpnbFile;
import org.jetbrains.plugins.ipnb.format.IpnbParser;
import org.jetbrains.plugins.ipnb.format.cells.IpnbCodeCell;
import java.io.IOException;
-public class CellOperationTest extends TestCase {
+public class CellOperationTest extends IpnbTestCase {
public void testAddCell() throws IOException {
final String fileName = "testData/emptyFile.ipynb";
- final String fileText = IpnbTestUtil.getFileText(fileName);
+ final String fileText = IpnbTestCase.getFileText(fileName);
final IpnbFile ipnbFile = IpnbParser.parseIpnbFile(fileText, new LightVirtualFile());
ipnbFile.addCell(IpnbCodeCell.createEmptyCodeCell(), ipnbFile.getCells().size());
public void testRemoveCell() throws IOException {
final String fileName = "testData/emptyFile.ipynb";
- final String fileText = IpnbTestUtil.getFileText(fileName);
+ final String fileText = IpnbTestCase.getFileText(fileName);
final IpnbFile ipnbFile = IpnbParser.parseIpnbFile(fileText, new LightVirtualFile());
ipnbFile.addCell(IpnbCodeCell.createEmptyCodeCell(), ipnbFile.getCells().size());
ipnbFile.removeCell(ipnbFile.getCells().size() - 1);
- assertEquals(fileText, IpnbTestUtil.getFileText(fileName));
+ assertEquals(fileText, IpnbTestCase.getFileText(fileName));
}
}
import com.intellij.openapi.application.PathManager;
+import junit.framework.TestCase;
import org.jetbrains.annotations.NotNull;
import java.io.BufferedReader;
import java.io.FileReader;
import java.io.IOException;
-public class IpnbTestUtil {
+public abstract class IpnbTestCase extends TestCase {
+
static String getFileText(@NotNull final String fileName) throws IOException {
String fullPath = PathManager.getHomePath() + "/community/python/ipnb/" + fileName;
final BufferedReader br = new BufferedReader(new FileReader(fullPath));
import com.google.common.collect.Iterables;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.testFramework.LightVirtualFile;
-import junit.framework.TestCase;
import org.jetbrains.plugins.ipnb.format.IpnbFile;
import org.jetbrains.plugins.ipnb.format.IpnbParser;
import org.jetbrains.plugins.ipnb.format.cells.IpnbCell;
import java.io.IOException;
import java.util.List;
-public class JsonParserTest extends TestCase {
+public class JsonParserTest extends IpnbTestCase {
public void testFile() throws IOException {
final String fileName = "testData/SymPy.ipynb";
- final String fileText = IpnbTestUtil.getFileText(fileName);
+ final String fileText = IpnbTestCase.getFileText(fileName);
final IpnbFile ipnbFile = IpnbParser.parseIpnbFile(fileText, new LightVirtualFile());
assertNotNull(ipnbFile);
public void testMarkdownCells() throws IOException {
final String fileName = "testData/SymPy.ipynb";
- final String fileText = IpnbTestUtil.getFileText(fileName);
+ final String fileText = IpnbTestCase.getFileText(fileName);
final IpnbFile ipnbFile = IpnbParser.parseIpnbFile(fileText, new LightVirtualFile());
assertNotNull(ipnbFile);
final List<IpnbCell> cells = ipnbFile.getCells();
public void testMarkdownCell() throws IOException {
final String fileName = "testData/markdown.ipynb";
- final String fileText = IpnbTestUtil.getFileText(fileName);
+ final String fileText = IpnbTestCase.getFileText(fileName);
final IpnbFile ipnbFile = IpnbParser.parseIpnbFile(fileText, new LightVirtualFile());
assertNotNull(ipnbFile);
final List<IpnbCell> cells = ipnbFile.getCells();
public void testCodeCell() throws IOException {
final String fileName = "testData/code.ipynb";
- final String fileText = IpnbTestUtil.getFileText(fileName);
+ final String fileText = IpnbTestCase.getFileText(fileName);
final IpnbFile ipnbFile = IpnbParser.parseIpnbFile(fileText, new LightVirtualFile());
assertNotNull(ipnbFile);
final List<IpnbCell> cells = ipnbFile.getCells();
public void testOutputs() throws IOException {
final String fileName = "testData/outputs.ipynb";
- final String fileText = IpnbTestUtil.getFileText(fileName);
+ final String fileText = IpnbTestCase.getFileText(fileName);
final IpnbFile ipnbFile = IpnbParser.parseIpnbFile(fileText, new LightVirtualFile());
assertNotNull(ipnbFile);
final List<IpnbCell> cells = ipnbFile.getCells();