diff -r b758351d191f -r cc9b7e14412b web/lib/django/contrib/gis/tests/geoapp/tests_mysql.py --- a/web/lib/django/contrib/gis/tests/geoapp/tests_mysql.py Wed May 19 17:43:59 2010 +0200 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,186 +0,0 @@ -""" - A limited test module is used for a limited spatial database. -""" -import os, unittest -from models import Country, City, State, Feature -from django.contrib.gis import gdal -from django.contrib.gis.geos import * -from django.core.exceptions import ImproperlyConfigured - -class GeoModelTest(unittest.TestCase): - - def test01_initial_sql(self): - "Testing geographic initial SQL." - # Ensuring that data was loaded from initial SQL. - self.assertEqual(2, Country.objects.count()) - self.assertEqual(8, City.objects.count()) - self.assertEqual(2, State.objects.count()) - - def test02_proxy(self): - "Testing Lazy-Geometry support (using the GeometryProxy)." - #### Testing on a Point - pnt = Point(0, 0) - nullcity = City(name='NullCity', point=pnt) - nullcity.save() - - # Making sure TypeError is thrown when trying to set with an - # incompatible type. - for bad in [5, 2.0, LineString((0, 0), (1, 1))]: - try: - nullcity.point = bad - except TypeError: - pass - else: - self.fail('Should throw a TypeError') - - # Now setting with a compatible GEOS Geometry, saving, and ensuring - # the save took, notice no SRID is explicitly set. - new = Point(5, 23) - nullcity.point = new - - # Ensuring that the SRID is automatically set to that of the - # field after assignment, but before saving. - self.assertEqual(4326, nullcity.point.srid) - nullcity.save() - - # Ensuring the point was saved correctly after saving - self.assertEqual(new, City.objects.get(name='NullCity').point) - - # Setting the X and Y of the Point - nullcity.point.x = 23 - nullcity.point.y = 5 - # Checking assignments pre & post-save. - self.assertNotEqual(Point(23, 5), City.objects.get(name='NullCity').point) - nullcity.save() - self.assertEqual(Point(23, 5), City.objects.get(name='NullCity').point) - nullcity.delete() - - #### Testing on a Polygon - shell = LinearRing((0, 0), (0, 100), (100, 100), (100, 0), (0, 0)) - inner = LinearRing((40, 40), (40, 60), (60, 60), (60, 40), (40, 40)) - - # Creating a State object using a built Polygon - ply = Polygon(shell, inner) - nullstate = State(name='NullState', poly=ply) - self.assertEqual(4326, nullstate.poly.srid) # SRID auto-set from None - nullstate.save() - - ns = State.objects.get(name='NullState') - self.assertEqual(ply, ns.poly) - - # Testing the `ogr` and `srs` lazy-geometry properties. - if gdal.HAS_GDAL: - self.assertEqual(True, isinstance(ns.poly.ogr, gdal.OGRGeometry)) - self.assertEqual(ns.poly.wkb, ns.poly.ogr.wkb) - self.assertEqual(True, isinstance(ns.poly.srs, gdal.SpatialReference)) - self.assertEqual('WGS 84', ns.poly.srs.name) - - # Changing the interior ring on the poly attribute. - new_inner = LinearRing((30, 30), (30, 70), (70, 70), (70, 30), (30, 30)) - ns.poly[1] = new_inner - ply[1] = new_inner - self.assertEqual(4326, ns.poly.srid) - ns.save() - self.assertEqual(ply, State.objects.get(name='NullState').poly) - ns.delete() - - def test03_contains_contained(self): - "Testing the 'contained', 'contains', and 'bbcontains' lookup types." - # Getting Texas, yes we were a country -- once ;) - texas = Country.objects.get(name='Texas') - - # Seeing what cities are in Texas, should get Houston and Dallas, - # and Oklahoma City because MySQL 'within' only checks on the - # _bounding box_ of the Geometries. - qs = City.objects.filter(point__within=texas.mpoly) - self.assertEqual(3, qs.count()) - cities = ['Houston', 'Dallas', 'Oklahoma City'] - for c in qs: self.assertEqual(True, c.name in cities) - - # Pulling out some cities. - houston = City.objects.get(name='Houston') - wellington = City.objects.get(name='Wellington') - pueblo = City.objects.get(name='Pueblo') - okcity = City.objects.get(name='Oklahoma City') - lawrence = City.objects.get(name='Lawrence') - - # Now testing contains on the countries using the points for - # Houston and Wellington. - tx = Country.objects.get(mpoly__contains=houston.point) # Query w/GEOSGeometry - nz = Country.objects.get(mpoly__contains=wellington.point.hex) # Query w/EWKBHEX - ks = State.objects.get(poly__contains=lawrence.point) - self.assertEqual('Texas', tx.name) - self.assertEqual('New Zealand', nz.name) - self.assertEqual('Kansas', ks.name) - - # Pueblo is not contained in Texas or New Zealand. - self.assertEqual(0, len(Country.objects.filter(mpoly__contains=pueblo.point))) # Query w/GEOSGeometry object - - # OK City is contained w/in bounding box of Texas. - qs = Country.objects.filter(mpoly__bbcontains=okcity.point) - self.assertEqual(1, len(qs)) - self.assertEqual('Texas', qs[0].name) - - def test04_disjoint(self): - "Testing the `disjoint` lookup type." - ptown = City.objects.get(name='Pueblo') - qs1 = City.objects.filter(point__disjoint=ptown.point) - self.assertEqual(7, qs1.count()) - # TODO: This query should work in MySQL, but it appears the - # `MBRDisjoint` function doesn't work properly (I went down - # to the SQL level for debugging and still got bogus answers). - #qs2 = State.objects.filter(poly__disjoint=ptown.point) - #self.assertEqual(1, qs2.count()) - #self.assertEqual('Kansas', qs2[0].name) - - def test05_equals(self): - "Testing the 'same_as' and 'equals' lookup types." - pnt = fromstr('POINT (-95.363151 29.763374)', srid=4326) - c1 = City.objects.get(point=pnt) - c2 = City.objects.get(point__same_as=pnt) - c3 = City.objects.get(point__equals=pnt) - for c in [c1, c2, c3]: self.assertEqual('Houston', c.name) - - def test06_geometryfield(self): - "Testing GeometryField." - f1 = Feature(name='Point', geom=Point(1, 1)) - f2 = Feature(name='LineString', geom=LineString((0, 0), (1, 1), (5, 5))) - f3 = Feature(name='Polygon', geom=Polygon(LinearRing((0, 0), (0, 5), (5, 5), (5, 0), (0, 0)))) - f4 = Feature(name='GeometryCollection', - geom=GeometryCollection(Point(2, 2), LineString((0, 0), (2, 2)), - Polygon(LinearRing((0, 0), (0, 5), (5, 5), (5, 0), (0, 0))))) - f1.save() - f2.save() - f3.save() - f4.save() - - f_1 = Feature.objects.get(name='Point') - self.assertEqual(True, isinstance(f_1.geom, Point)) - self.assertEqual((1.0, 1.0), f_1.geom.tuple) - f_2 = Feature.objects.get(name='LineString') - self.assertEqual(True, isinstance(f_2.geom, LineString)) - self.assertEqual(((0.0, 0.0), (1.0, 1.0), (5.0, 5.0)), f_2.geom.tuple) - - f_3 = Feature.objects.get(name='Polygon') - self.assertEqual(True, isinstance(f_3.geom, Polygon)) - f_4 = Feature.objects.get(name='GeometryCollection') - self.assertEqual(True, isinstance(f_4.geom, GeometryCollection)) - self.assertEqual(f_3.geom, f_4.geom[2]) - - def test07_mysql_limitations(self): - "Testing that union(), kml(), gml() raise exceptions." - self.assertRaises(ImproperlyConfigured, City.objects.union, Point(5, 23), field_name='point') - self.assertRaises(ImproperlyConfigured, State.objects.all().kml, field_name='poly') - self.assertRaises(ImproperlyConfigured, Country.objects.all().gml, field_name='mpoly') - -from test_feeds import GeoFeedTest -from test_regress import GeoRegressionTests -from test_sitemaps import GeoSitemapTest - -def suite(): - s = unittest.TestSuite() - s.addTest(unittest.makeSuite(GeoModelTest)) - s.addTest(unittest.makeSuite(GeoFeedTest)) - s.addTest(unittest.makeSuite(GeoSitemapTest)) - s.addTest(unittest.makeSuite(GeoRegressionTests)) - return s