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)