Added equality tests for #63

stchris · Nov 18, 2019 · f38ffb1 · f38ffb1
1 parent 050c1d4
commit f38ffb1
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/tests.py b/tests/tests.py
@@ -366,6 +366,21 @@ def test_invalid_external_dtd(self):
             untangle.parse(self.bad_dtd_xml, feature_external_ges=True)
 
 
+class TestEquals(unittest.TestCase):
+
+    def test_equals(self):
+        a = untangle.Element('a', '1')
+        b = untangle.Element('b', '1')
+        self.assertTrue(a == b)
+
+    def test_list_equals(self):
+        a = untangle.Element('a', '1')
+        b = untangle.Element('b', '1')
+        listA = [a, b]
+        c = untangle.Element('c', '1')
+        self.assertTrue(c in listA)
+
+
 if __name__ == "__main__":
     unittest.main()