def test_parse_codeblock(self): b = CodeBlock() b.add_dependency("test", "blah") b.write_line("if 1:") b.write_line("do()", 1) n, v = parse_code(""" if 2: $doit """, None, doit=b) self.assertEqual(str(n), "if 2:\n if 1:\n do()") self.assertTrue("test" in n.get_dependencies())
def test_codeblock_print(self): f = StringIO() a = CodeBlock() a.write_line("abc") a.pprint(f) self.assertEqual(f.getvalue(), "abc\n")