22from io import StringIO
33from unittest .mock import patch
44
5- from django import VERSION
65from django .conf import settings
76from django .contrib .gis import gdal
87from django .contrib .gis .db .models import GeometryField
1312
1413from mapentity .registry import app_settings
1514from mapentity .serializers import ZipShapeSerializer , CSVSerializer
16- from . .models import MushroomSpot , Tag , Supermarket
15+ from test_app .models import MushroomSpot , Tag , Supermarket
1716
1817
1918class CommonShapefileSerializerMixin (object ):
@@ -28,19 +27,21 @@ def getShapefileLayers(self):
2827
2928class MushroomShapefileSerializerTest (CommonShapefileSerializerMixin , TestCase ):
3029 def setUp (self ):
31- self .point1 = MushroomSpot .objects .create (serialized = 'SRID=%s ;POINT(0 0)' % settings . SRID )
30+ self .point1 = MushroomSpot .objects .create (serialized = f 'SRID={ settings . SRID } ;POINT(0 0)' )
3231 self .point1 .tags .add (Tag .objects .create (label = "Tag1" ))
3332 self .point1 .tags .add (Tag .objects .create (label = "Tag2" ))
34- self .line1 = MushroomSpot .objects .create (serialized = 'SRID=%s;LINESTRING(0 0, 10 0)' % settings .SRID )
35- self .multipoint = MushroomSpot .objects .create (serialized = 'SRID=%s;MULTIPOINT((1 1), (2 2))' % settings .SRID )
36- self .multiline = MushroomSpot .objects .create (serialized = 'SRID=%s;MULTILINESTRING((1 1, 2 2), '
37- '(3 3, 4 4))' % settings .SRID )
38- self .polygon = MushroomSpot .objects .create (serialized = 'SRID=%s;POLYGON((1 1, 2 2, '
39- '1 2, 1 1))' % settings .SRID )
40- self .multipolygon = MushroomSpot .objects .create (serialized = 'SRID=%s;MULTIPOLYGON(((1 1, 2 2, '
41- '1 2, 1 1)))' % settings .SRID )
33+ self .line1 = MushroomSpot .objects .create (serialized = f'SRID={ settings .SRID } ;LINESTRING(0 0, 10 0)' )
34+ self .multipoint = MushroomSpot .objects .create (serialized = f'SRID={ settings .SRID } ;MULTIPOINT((1 1), (2 2))' )
35+ self .multiline = MushroomSpot .objects .create (
36+ serialized = f'SRID={ settings .SRID } ;MULTILINESTRING((1 1, 2 2), (3 3, 4 4))'
37+ )
38+ self .polygon = MushroomSpot .objects .create (serialized = f'SRID={ settings .SRID } ;POLYGON((1 1, 2 2, 1 2, 1 1))' )
39+ self .multipolygon = MushroomSpot .objects .create (
40+ serialized = f'SRID={ settings .SRID } ;MULTIPOLYGON(((1 1, 2 2, 1 2, 1 1)))'
41+ )
4242 self .geometrycollection = MushroomSpot .objects .create (
43- serialized = 'SRID=%s;GEOMETRYCOLLECTION(POINT(0 0), POLYGON((1 1, 2 2, 1 2, 1 1))))' % settings .SRID )
43+ serialized = f'SRID={ settings .SRID } ;GEOMETRYCOLLECTION(POINT(0 0), POLYGON((1 1, 2 2, 1 2, 1 1))))'
44+ )
4445 MushroomSpot .geomfield = GeometryField (name = 'geom' , srid = settings .SRID )
4546
4647 self .serializer = ZipShapeSerializer ()
@@ -188,10 +189,7 @@ def test_serializer_foreign_key(self):
188189 layers = self .getShapefileLayers ()
189190 layer = layers ['Polygon.shp' ]
190191 feature = layer [0 ]
191- if VERSION [0 ] >= 3 :
192- self .assertEqual (feature ['tag' ].value , None )
193- else :
194- self .assertEqual (feature ['tag' ].value , "" )
192+ self .assertEqual (feature ['tag' ].value , None )
195193
196194 self .serializer = ZipShapeSerializer ()
197195 tag = Tag .objects .create (label = "Tag" )
0 commit comments