def test_insert(self): pxml = PXML(4) pxml.insert("Hello") self.assertEqual(["Hello"], pxml.raw) self.assertEqual("Hello", str(pxml)) pxml.insert("World") self.assertEqual(["Hello", "World"], pxml.raw) self.assertEqual("HelloWorld", str(pxml)) pxml.depth = 1 pxml.indent().insert("One") one = " " * pxml.spaces * pxml.depth self.assertEqual(["Hello", "World", one, "One"], pxml.raw) self.assertEqual("HelloWorld{}One".format(one), str(pxml)) pxml.depth = 2 pxml.indent().insert("Two") two = " " * pxml.spaces * pxml.depth self.assertEqual(["Hello", "World", one, "One", two, "Two"], pxml.raw) self.assertEqual("HelloWorld{}One{}Two".format(one, two), str(pxml)) pxml.depth = 3 pxml.indent().insert("Three") three = " " * pxml.spaces * pxml.depth self.assertEqual(["Hello", "World", one, "One", two, "Two", three, "Three"], pxml.raw) self.assertEqual("HelloWorld{}One{}Two{}Three".format(one, two, three), str(pxml)) pxml.insert("Bye").insert("Bye") self.assertEqual(["Hello", "World", one, "One", two, "Two", three, "Three", "Bye", "Bye"], pxml.raw) self.assertEqual("HelloWorld{}One{}Two{}ThreeByeBye".format(one, two, three), str(pxml))
def test_indent(self): pxml = PXML() pxml.indent() self.assertEqual([], pxml.raw) self.assertEqual("", str(pxml)) pxml = PXML(4) pxml.depth = 1 pxml.indent(5) self.assertEqual([" " * pxml.spaces] * 5, pxml.raw) self.assertEqual(" " * pxml.spaces * 5, str(pxml)) for spaces in range(1, 5): for depth in range(1, 5): pxml = PXML(spaces) pxml.depth = depth pxml.indent() self.assertEqual([" " * spaces * depth], pxml.raw) self.assertEqual(" " * spaces * depth, str(pxml))