class TestPrinter(TestCase): def setUp(self): self.paper = Paper() self.printer = Printer() self.printer.insert(self.paper) def test_addIndentationAtTheStartOfALineAsLongAsItIsNotRemoved(self): driver = MethodDefinitionPrinterDriver(self.printer) tree = ast.parse(open("samples/method.py", 'r').read()) driver.visit(tree) self.assertEqual(self.paper.text, 'public Object basic_method( Object one, Object two){\n}')
class TestPrinter(TestCase): def setUp(self): self.paper = Paper() self.printer = Printer() self.printer.insert(self.paper) def test_justVisitAllChildrenIfNoSpecialPrinterIsRegistered(self): self.printer.write("Some text") self.assertEqual(self.paper.text, "Some text") def test_changeLine(self): self.printer.write("Some text") self.printer.newline() self.printer.write("more text") self.assertEqual(self.paper.text, "Some text\nmore text") def test_addIndentationAtTheStartOfALine(self): self.printer.increaseIndentation() self.printer.write("Some text") self.assertEqual(self.paper.text, " Some text") def test_decreaseIndentationLevelInNextLine(self): self.printer.increaseIndentation() self.printer.decreaseIndentation() self.printer.write("Some text") self.assertEqual(self.paper.text, "Some text") def test_doNotAllowNegativeIndentationLevels(self): self.printer.decreaseIndentation() self.printer.increaseIndentation() self.printer.write("Some text") self.assertEqual(self.paper.text, " Some text") def test_addIndentationAtTheStartOfALineAsLongAsItIsNotRemoved(self): self.printer.increaseIndentation() self.printer.write("Some text") self.printer.newline() self.printer.write("more text") self.assertEqual(self.paper.text, " Some text\n more text")