from unittest import TestCase
from distribution.models import import_model


class ImportModelTest(TestCase):

    def test_import_model_with_incorrect_path(self):
        self.assertRaises(
            ImportError,
            import_model,
            'wrongpath'
        )

    def test_import_model_with_incorrect_attribute_class(self):
        self.assertRaises(
            ImportError,
            import_model,
            'distribution.models.taxonomy.wrongattribute'
        )