diff --git a/tests/unit/factory/test_optable.py b/tests/unit/factory/test_optable.py index 4deb57e92..b81e7a517 100644 --- a/tests/unit/factory/test_optable.py +++ b/tests/unit/factory/test_optable.py @@ -117,11 +117,11 @@ def test_generate_sax_parser_item_with_many_slash(self): Loader=yaml.Loader))) tbl = taskmallocdetail(self.dev) data = generate_sax_parser_input(tbl) - self.assertEqual( - etree.tostring(data), + self.assertEqual(data.tag, 'task-memory-malloc-usage-report') + self.assertEqual(len(etree.tostring(data)), len( b'') + b'-calls/>')) def test_generate_sax_parser_same_parents_with_diff_fields(self): yaml_data = """ @@ -163,8 +163,8 @@ def test_generate_sax_parser_same_parents_with_diff_fields(self): Loader=yaml.Loader))) tbl = VtepTable(self.dev) data = generate_sax_parser_input(tbl) - self.assertEqual( - etree.tostring(data), + self.assertEqual(data.tag, 'physical-interface') + self.assertEqual(len(etree.tostring(data)),len( b'' b'' b'' @@ -172,7 +172,7 @@ def test_generate_sax_parser_same_parents_with_diff_fields(self): b'' b'' b'' - b'' + b'') ) def _read_file(self, fname):