From 109c019836c229ee317e4867def49ec2fbe9dde2 Mon Sep 17 00:00:00 2001
From: Blake Eggleston <bdeggleston@gmail.com>
Date: Thu, 20 Mar 2014 16:32:14 -0700
Subject: [PATCH] adding tests around equality

---
 cqlengine/tests/model/test_model.py | 33 +++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 cqlengine/tests/model/test_model.py

diff --git a/cqlengine/tests/model/test_model.py b/cqlengine/tests/model/test_model.py
new file mode 100644
index 00000000..a0a6929a
--- /dev/null
+++ b/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)