diff --git a/pinry/core/tests/helpers.py b/pinry/core/tests/helpers.py index f2ea741..e5a9030 100644 --- a/pinry/core/tests/helpers.py +++ b/pinry/core/tests/helpers.py @@ -7,6 +7,9 @@ from django.test import TestCase import factory from taggit.models import Tag +from ..models import Pin, Image +from ...users.models import User + TEST_IMAGE_PATH = settings.SITE_ROOT + 'screenshot.png' @@ -70,4 +73,4 @@ class PinFactoryTest(TestCase): def test_empty_tags(self): PinFactory(tags=[]) - self.assertEqual(Tag.objects.count(), 0) \ No newline at end of file + self.assertEqual(Tag.objects.count(), 0)