34 lines
903 B
Python
34 lines
903 B
Python
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)
|