def test_notebookCellSeparator(self):
result = self.__lineTransformer.transform(0, [
ImportLine('DataSentics.Greetings.Greeter', 'Greeter'),
CellSeparatorLine()
])
self.assertLines([
CellSeparatorLine(),
EmptyStringLine(),
MagicRunLine(ImportLine('DataSentics.Greetings.Greeter', 'Greeter'),)
], result)
def test_scriptImportEmptyLineImport(self):
result = self.__lineTransformer.transform(0, [
ImportLine('DataSentics.Greetings.Greeter', 'Greeter'),
EmptyStringLine(),
ImportLine('DataSentics.Greetings.Greeter2', 'Greeter2')
])
self.assertLines([
CellSeparatorLine(),
EmptyStringLine(),
MagicRunLine(ImportLine('DataSentics.Greetings.Greeter', 'Greeter'))
], result)
def test_scriptCellSeparatorBefore(self):
result = self.__lineTransformer.transform(2, [
CellSeparatorLine(),
EmptyStringLine(),
ImportLine('DataSentics.Greetings.Greeter', 'Greeter'),
StringLine('print("Hello world")')
])
self.assertLines([
MagicRunLine(ImportLine('DataSentics.Greetings.Greeter', 'Greeter')),
EmptyStringLine(),
CellSeparatorLine()
], result)
def parse(self, line):
matches = re.match('^from[ ]+([^ ]+) import ([^ ]+)$', line)
if matches is None:
return None
else:
return ImportLine(matches.group(1), matches.group(2))
def test_simple(self):
result = self.__transformer.transform([
ImportLine('DataSentics.Greetings.Greeter', 'Greeter'),
EmptyStringLine(),
StringLine('print("Hello world")'),
ImportLine('DataSentics.local_bootstrap', '*'),
EmptyStringLine(),
StringLine('a = 5'),
])
expected = [
ImportLine('DataSentics.Greetings.Greeter', 'Greeter'),
EmptyStringLine(),
StringLine('print("Hello world")'),
EmptyStringLine(),
StringLine('a = 5'),
]
self.assertLines(expected, result)
def test_multilineNotebook(self):
result = self.__linesTransformer.transform([
CellSeparatorLine(),
EmptyStringLine(),
ImportLine('DataSentics.Greetings.Greeter', 'Greeter'),
EmptyStringLine(),
ImportLine('Foo.Bar', '*'),
EmptyStringLine(),
StringLine('print("Hello world")'),
EmptyStringLine(),
CellSeparatorLine(),
ImportLine('Hello.World', '*'),
EmptyStringLine(),
CellSeparatorLine(),
StringLine('a = 5'),
])
expected = [
CellSeparatorLine(),
EmptyStringLine(),
MagicRunLine(ImportLine('DataSentics.Greetings.Greeter', 'Greeter')),
EmptyStringLine(),
CellSeparatorLine(),
EmptyStringLine(),
MagicRunLine(ImportLine('Foo.Bar', '*')),
EmptyStringLine(),
CellSeparatorLine(),
EmptyStringLine(), # TODO: chceme tedy empty string? Spíš ne
StringLine('print("Hello world")'),
EmptyStringLine(),
CellSeparatorLine(),
MagicRunLine(ImportLine('Hello.World', '*')),
EmptyStringLine(),
CellSeparatorLine(),
StringLine('a = 5'),
]
self.assertLines(expected, result)