def test2annotate_match(self): """This test case checks if all possible annotations are created correctly.""" a3 = MatchElement('a3', b'a3', b'a3', []) a2 = MatchElement('a2', b'a2', b'a2', [a3]) a1 = MatchElement('a1', b'a1', b'a1', [a2]) b3 = MatchElement('b3', b'b3', b'b3', []) b2 = MatchElement('b2', b'b2', b'b2', [b3]) b1 = MatchElement('b1', b'b1', b'b1', [b2]) root_element = MatchElement('root', b'root', b'root', [a1, b1]) self.assertEqual(root_element.annotate_match(None), "root: b'root' a1: b'a1' a2: b'a2' a3: b'a3' b1: b'b1' b2: b'b2' b3: b'b3'") self.assertEqual(root_element.annotate_match(''), "root: b'root'\n a1: b'a1'\n a2: b'a2'\n a3: b'a3'\n b1: b'b1'\n" " b2: b'b2'\n b3: b'b3'")
def test5annotate_match(self): """This test case checks if all possible annotations are created correctly.""" a3 = MatchElement("a3", b"a3", b"a3", None) a2 = MatchElement("a2", b"a2", b"a2", [a3]) a1 = MatchElement("a1", b"a1", b"a1", [a2]) b3 = MatchElement("b3", b"b3", b"b3", None) b2 = MatchElement("b2", b"b2", b"b2", [b3]) b1 = MatchElement("b1", b"b1", b"b1", [b2]) root_element = MatchElement("root", b"root", b"root", [a1, b1]) self.assertEqual( root_element.annotate_match(None), "root: root a1: a1 a2: a2 a3: a3 b1: b1 b2: b2 b3: b3") self.assertEqual( root_element.annotate_match(""), "root: root\n a1: a1\n a2: a2\n a3: a3\n b1: b1\n b2: b2\n " "b3: b3") self.assertEqual( root_element.annotate_match("--"), "--root: root\n-- a1: a1\n-- a2: a2\n-- a3: a3\n-- b1: b1\n" "-- b2: b2\n-- b3: b3")