adding tests around equality

cqlengine/tests/model/test_model.py

@@ -0,0 +1,33 @@
+from unittest import TestCase
+
+from cqlengine.models import Model
+from cqlengine import columns
+
+
+class TestModel(TestCase):
+    """ Tests the non-io functionality of models """
+
+    def test_instance_equality(self):
+        """ tests the model equality functionality """
+        class EqualityModel(Model):
+            pk = columns.Integer(primary_key=True)
+
+        m0 = EqualityModel(pk=0)
+        m1 = EqualityModel(pk=1)
+
+        self.assertEqual(m0, m0)
+        self.assertNotEqual(m0, m1)
+
+    def test_model_equality(self):
+        """ tests the model equality functionality """
+        class EqualityModel0(Model):
+            pk = columns.Integer(primary_key=True)
+
+        class EqualityModel1(Model):
+            kk = columns.Integer(primary_key=True)
+
+        m0 = EqualityModel0(pk=0)
+        m1 = EqualityModel1(kk=1)
+
+        self.assertEqual(m0, m0)
+        self.assertNotEqual(m0, m1)