"""
This module cantains tests for the general Kontor data model.
"""
from . import TestBase
from kontor.models import User


class TestKontorModel(TestBase):
    """This TestCase contains tests for users."""
    def test_user_model(self):
        """
        Test number of records in User collection
        """
        self.assertEqual(User.objects.all().count(), 2)