diff -r 000000000000 -r 0d40e90630ef web/lib/django/contrib/gis/geos/tests/test_io.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/web/lib/django/contrib/gis/geos/tests/test_io.py Wed Jan 20 00:34:04 2010 +0100 @@ -0,0 +1,112 @@ +import binascii, ctypes, unittest +from django.contrib.gis.geos import GEOSGeometry, WKTReader, WKTWriter, WKBReader, WKBWriter, geos_version_info + +class GEOSIOTest(unittest.TestCase): + + def test01_wktreader(self): + # Creating a WKTReader instance + wkt_r = WKTReader() + wkt = 'POINT (5 23)' + + # read() should return a GEOSGeometry + ref = GEOSGeometry(wkt) + g1 = wkt_r.read(wkt) + g2 = wkt_r.read(unicode(wkt)) + + for geom in (g1, g2): + self.assertEqual(ref, geom) + + # Should only accept basestring objects. + self.assertRaises(TypeError, wkt_r.read, 1) + self.assertRaises(TypeError, wkt_r.read, buffer('foo')) + + def test02_wktwriter(self): + # Creating a WKTWriter instance, testing its ptr property. + wkt_w = WKTWriter() + self.assertRaises(TypeError, wkt_w._set_ptr, WKTReader.ptr_type()) + + ref = GEOSGeometry('POINT (5 23)') + ref_wkt = 'POINT (5.0000000000000000 23.0000000000000000)' + self.assertEqual(ref_wkt, wkt_w.write(ref)) + + def test03_wkbreader(self): + # Creating a WKBReader instance + wkb_r = WKBReader() + + hex = '000000000140140000000000004037000000000000' + wkb = buffer(binascii.a2b_hex(hex)) + ref = GEOSGeometry(hex) + + # read() should return a GEOSGeometry on either a hex string or + # a WKB buffer. + g1 = wkb_r.read(wkb) + g2 = wkb_r.read(hex) + for geom in (g1, g2): + self.assertEqual(ref, geom) + + bad_input = (1, 5.23, None, False) + for bad_wkb in bad_input: + self.assertRaises(TypeError, wkb_r.read, bad_wkb) + + def test04_wkbwriter(self): + wkb_w = WKBWriter() + + # Representations of 'POINT (5 23)' in hex -- one normal and + # the other with the byte order changed. + g = GEOSGeometry('POINT (5 23)') + hex1 = '010100000000000000000014400000000000003740' + wkb1 = buffer(binascii.a2b_hex(hex1)) + hex2 = '000000000140140000000000004037000000000000' + wkb2 = buffer(binascii.a2b_hex(hex2)) + + self.assertEqual(hex1, wkb_w.write_hex(g)) + self.assertEqual(wkb1, wkb_w.write(g)) + + # Ensuring bad byteorders are not accepted. + for bad_byteorder in (-1, 2, 523, 'foo', None): + # Equivalent of `wkb_w.byteorder = bad_byteorder` + self.assertRaises(ValueError, wkb_w._set_byteorder, bad_byteorder) + + # Setting the byteorder to 0 (for Big Endian) + wkb_w.byteorder = 0 + self.assertEqual(hex2, wkb_w.write_hex(g)) + self.assertEqual(wkb2, wkb_w.write(g)) + + # Back to Little Endian + wkb_w.byteorder = 1 + + # Now, trying out the 3D and SRID flags. + g = GEOSGeometry('POINT (5 23 17)') + g.srid = 4326 + + hex3d = '0101000080000000000000144000000000000037400000000000003140' + wkb3d = buffer(binascii.a2b_hex(hex3d)) + hex3d_srid = '01010000A0E6100000000000000000144000000000000037400000000000003140' + wkb3d_srid = buffer(binascii.a2b_hex(hex3d_srid)) + + # Ensuring bad output dimensions are not accepted + for bad_outdim in (-1, 0, 1, 4, 423, 'foo', None): + # Equivalent of `wkb_w.outdim = bad_outdim` + self.assertRaises(ValueError, wkb_w._set_outdim, bad_outdim) + + # These tests will fail on 3.0.0 because of a bug that was fixed in 3.1: + # http://trac.osgeo.org/geos/ticket/216 + if not geos_version_info()['version'].startswith('3.0.'): + # Now setting the output dimensions to be 3 + wkb_w.outdim = 3 + + self.assertEqual(hex3d, wkb_w.write_hex(g)) + self.assertEqual(wkb3d, wkb_w.write(g)) + + # Telling the WKBWriter to inlcude the srid in the representation. + wkb_w.srid = True + self.assertEqual(hex3d_srid, wkb_w.write_hex(g)) + self.assertEqual(wkb3d_srid, wkb_w.write(g)) + +def suite(): + s = unittest.TestSuite() + s.addTest(unittest.makeSuite(GEOSIOTest)) + return s + +def run(verbosity=2): + unittest.TextTestRunner(verbosity=verbosity).run(suite())