Merge pull request #67 from stchris/equals-tests

Added equality tests for #63
stchris · Nov 18, 2019 · f8be900 · f8be900
2 parents 050c1d4 + f38ffb1
commit f8be900
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()