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")