From 2261875d3c64c00d449bbe988e0f6ccf8a4309ee Mon Sep 17 00:00:00 2001
From: littlewolf <nickptic@code-artel.com>
Date: Tue, 24 Sep 2019 21:36:59 +0300
Subject: [PATCH] Fix tests

---
 apps/location/tests.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/apps/location/tests.py b/apps/location/tests.py
index d85db414..f68ba56b 100644
--- a/apps/location/tests.py
+++ b/apps/location/tests.py
@@ -5,7 +5,6 @@ from account.models import User
 from rest_framework import status
 from http.cookies import SimpleCookie
 
-from django.contrib.gis.db.models import PointField
 from location.models import City, Region, Country
 
 
@@ -93,11 +92,16 @@ class RegionTests(BaseTestCase):
         self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
 
 
-class CityTests(RegionTests):
+class CityTests(BaseTestCase):
 
     def setUp(self):
         super().setUp()
 
+        self.country = Country.objects.create(
+            name=json.dumps({"en-GB": "Test country"}),
+            code="test"
+        )
+
         self.region = Region.objects.create(
             name="Test region",
             code="812",
@@ -137,6 +141,7 @@ class AddressTests(BaseTestCase):
 
     def setUp(self):
         super().setUp()
+
         self.country = Country.objects.create(
             name=json.dumps({"en-GB": "Test country"}),
             code="test"