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_deps(self):
a = CodeBlock()
a.add_dependency("a", object())
self.assertRaises(ValueError, a.add_dependency, "a", object())
def test_codeblock_print(self):
f = StringIO()
a = CodeBlock()
a.write_line("abc")
a.pprint(f)
self.assertEqual(f.getvalue(), "abc\n")
def test_codeblock(self):
a = CodeBlock("foo")
self.assertEqual(str(a), "foo")
a.write_lines(["1", "2"], 0)
self.assertEqual(str(a), "foo\n1\n2")
