diff --git a/wagtailflags/tests/test_conditions.py b/wagtailflags/tests/test_conditions.py index 67b80be..1a8ac07 100644 --- a/wagtailflags/tests/test_conditions.py +++ b/wagtailflags/tests/test_conditions.py @@ -1,9 +1,10 @@ from django.test import RequestFactory, TestCase import wagtail -from flags.conditions import RequiredForCondition from wagtail.core.models import Site +from flags.conditions import RequiredForCondition + from wagtailflags.conditions import site_condition diff --git a/wagtailflags/tests/test_signals.py b/wagtailflags/tests/test_signals.py index c5ebb71..09e451b 100644 --- a/wagtailflags/tests/test_signals.py +++ b/wagtailflags/tests/test_signals.py @@ -2,9 +2,10 @@ from django.test import TestCase -from flags.models import FlagState from wagtail.tests.utils import WagtailTestUtils +from flags.models import FlagState + from wagtailflags.signals import flag_disabled, flag_enabled diff --git a/wagtailflags/tests/test_views.py b/wagtailflags/tests/test_views.py index bacc69c..3f4328f 100644 --- a/wagtailflags/tests/test_views.py +++ b/wagtailflags/tests/test_views.py @@ -1,8 +1,9 @@ from django.test import TestCase -from flags.models import FlagState from wagtail.tests.utils import WagtailTestUtils +from flags.models import FlagState + class TestWagtailFlagsViews(TestCase, WagtailTestUtils): def setUp(self):