56 lines
2.4 KiB
Python
56 lines
2.4 KiB
Python
from datetime import datetime
|
|
from django.contrib.gis.tests.utils import no_mysql, no_spatialite
|
|
from django.contrib.gis.shortcuts import render_to_kmz
|
|
from django.test import TestCase
|
|
from models import City, PennsylvaniaCity, State
|
|
|
|
class GeoRegressionTests(TestCase):
|
|
|
|
def test01_update(self):
|
|
"Testing GeoQuerySet.update(), see #10411."
|
|
pnt = City.objects.get(name='Pueblo').point
|
|
bak = pnt.clone()
|
|
pnt.y += 0.005
|
|
pnt.x += 0.005
|
|
|
|
City.objects.filter(name='Pueblo').update(point=pnt)
|
|
self.assertEqual(pnt, City.objects.get(name='Pueblo').point)
|
|
City.objects.filter(name='Pueblo').update(point=bak)
|
|
self.assertEqual(bak, City.objects.get(name='Pueblo').point)
|
|
|
|
def test02_kmz(self):
|
|
"Testing `render_to_kmz` with non-ASCII data, see #11624."
|
|
name = '\xc3\x85land Islands'.decode('iso-8859-1')
|
|
places = [{'name' : name,
|
|
'description' : name,
|
|
'kml' : '<Point><coordinates>5.0,23.0</coordinates></Point>'
|
|
}]
|
|
kmz = render_to_kmz('gis/kml/placemarks.kml', {'places' : places})
|
|
|
|
@no_spatialite
|
|
@no_mysql
|
|
def test03_extent(self):
|
|
"Testing `extent` on a table with a single point, see #11827."
|
|
pnt = City.objects.get(name='Pueblo').point
|
|
ref_ext = (pnt.x, pnt.y, pnt.x, pnt.y)
|
|
extent = City.objects.filter(name='Pueblo').extent()
|
|
for ref_val, val in zip(ref_ext, extent):
|
|
self.assertAlmostEqual(ref_val, val, 4)
|
|
|
|
def test04_unicode_date(self):
|
|
"Testing dates are converted properly, even on SpatiaLite, see #16408."
|
|
founded = datetime(1857, 5, 23)
|
|
mansfield = PennsylvaniaCity.objects.create(name='Mansfield', county='Tioga', point='POINT(-77.071445 41.823881)',
|
|
founded=founded)
|
|
self.assertEqual(founded, PennsylvaniaCity.objects.dates('founded', 'day')[0])
|
|
|
|
def test05_empty_count(self):
|
|
"Testing that PostGISAdapter.__eq__ does check empty strings, see #13670"
|
|
# contrived example, but need a geo lookup paired with an id__in lookup
|
|
pueblo = City.objects.get(name='Pueblo')
|
|
state = State.objects.filter(poly__contains=pueblo.point)
|
|
cities_within_state = City.objects.filter(id__in=state)
|
|
|
|
# .count() should not throw TypeError in __eq__
|
|
self.assertEqual(cities_within_state.count(), 1)
|