Sanitized test assertions in gdal tests

This commit is contained in:
Claude Paroz 2013-10-10 23:01:07 +02:00
parent 1dae4ac177
commit 4b0f8d23c9
1 changed files with 23 additions and 22 deletions

View File

@ -38,19 +38,20 @@ class OGRGeomTest(unittest.TestCase, TestDataMixin):
self.assertRaises(OGRException, OGRGeomType, 9) self.assertRaises(OGRException, OGRGeomType, 9)
# Equivalence can take strings, ints, and other OGRGeomTypes # Equivalence can take strings, ints, and other OGRGeomTypes
self.assertEqual(True, OGRGeomType(1) == OGRGeomType(1)) self.assertEqual(OGRGeomType(1), OGRGeomType(1))
self.assertEqual(True, OGRGeomType(7) == 'GeometryCollection') self.assertEqual(OGRGeomType(7), 'GeometryCollection')
self.assertEqual(True, OGRGeomType('point') == 'POINT') self.assertEqual(OGRGeomType('point'), 'POINT')
self.assertEqual(False, OGRGeomType('point') == 2) self.assertNotEqual(OGRGeomType('point'), 2)
self.assertEqual(True, OGRGeomType('unknown') == 0) self.assertEqual(OGRGeomType('unknown'), 0)
self.assertEqual(True, OGRGeomType(6) == 'MULtiPolyGON') self.assertEqual(OGRGeomType(6), 'MULtiPolyGON')
self.assertEqual(False, OGRGeomType(1) != OGRGeomType('point')) self.assertEqual(OGRGeomType(1), OGRGeomType('point'))
self.assertEqual(True, OGRGeomType('POINT') != OGRGeomType(6)) self.assertNotEqual(OGRGeomType('POINT'), OGRGeomType(6))
# Testing the Django field name equivalent property. # Testing the Django field name equivalent property.
self.assertEqual('PointField', OGRGeomType('Point').django) self.assertEqual('PointField', OGRGeomType('Point').django)
self.assertEqual('GeometryField', OGRGeomType('Geometry').django)
self.assertEqual('GeometryField', OGRGeomType('Unknown').django) self.assertEqual('GeometryField', OGRGeomType('Unknown').django)
self.assertEqual(None, OGRGeomType('none').django) self.assertIsNone(OGRGeomType('none').django)
# 'Geometry' initialization implies an unknown geometry type. # 'Geometry' initialization implies an unknown geometry type.
gt = OGRGeomType('Geometry') gt = OGRGeomType('Geometry')
@ -60,8 +61,8 @@ class OGRGeomTest(unittest.TestCase, TestDataMixin):
def test00b_geomtype_25d(self): def test00b_geomtype_25d(self):
"Testing OGRGeomType object with 25D types." "Testing OGRGeomType object with 25D types."
wkb25bit = OGRGeomType.wkb25bit wkb25bit = OGRGeomType.wkb25bit
self.assertTrue(OGRGeomType(wkb25bit + 1) == 'Point25D') self.assertEqual(OGRGeomType(wkb25bit + 1), 'Point25D')
self.assertTrue(OGRGeomType('MultiLineString25D') == (5 + wkb25bit)) self.assertEqual(OGRGeomType('MultiLineString25D'), (5 + wkb25bit))
self.assertEqual('GeometryCollectionField', OGRGeomType('GeometryCollection25D').django) self.assertEqual('GeometryCollectionField', OGRGeomType('GeometryCollection25D').django)
def test01a_wkt(self): def test01a_wkt(self):
@ -159,8 +160,8 @@ class OGRGeomTest(unittest.TestCase, TestDataMixin):
self.assertEqual('LINESTRING', linestr.geom_name) self.assertEqual('LINESTRING', linestr.geom_name)
self.assertEqual(ls.n_p, linestr.point_count) self.assertEqual(ls.n_p, linestr.point_count)
self.assertEqual(ls.coords, linestr.tuple) self.assertEqual(ls.coords, linestr.tuple)
self.assertEqual(True, linestr == OGRGeometry(ls.wkt)) self.assertEqual(linestr, OGRGeometry(ls.wkt))
self.assertEqual(True, linestr != prev) self.assertNotEqual(linestr, prev)
self.assertRaises(OGRIndexError, linestr.__getitem__, len(linestr)) self.assertRaises(OGRIndexError, linestr.__getitem__, len(linestr))
prev = linestr prev = linestr
@ -179,8 +180,8 @@ class OGRGeomTest(unittest.TestCase, TestDataMixin):
self.assertEqual('MULTILINESTRING', mlinestr.geom_name) self.assertEqual('MULTILINESTRING', mlinestr.geom_name)
self.assertEqual(mls.n_p, mlinestr.point_count) self.assertEqual(mls.n_p, mlinestr.point_count)
self.assertEqual(mls.coords, mlinestr.tuple) self.assertEqual(mls.coords, mlinestr.tuple)
self.assertEqual(True, mlinestr == OGRGeometry(mls.wkt)) self.assertEqual(mlinestr, OGRGeometry(mls.wkt))
self.assertEqual(True, mlinestr != prev) self.assertNotEqual(mlinestr, prev)
prev = mlinestr prev = mlinestr
for ls in mlinestr: for ls in mlinestr:
self.assertEqual(2, ls.geom_type) self.assertEqual(2, ls.geom_type)
@ -195,8 +196,8 @@ class OGRGeomTest(unittest.TestCase, TestDataMixin):
#self.assertEqual(101, lr.geom_type.num) #self.assertEqual(101, lr.geom_type.num)
self.assertEqual('LINEARRING', lr.geom_name) self.assertEqual('LINEARRING', lr.geom_name)
self.assertEqual(rr.n_p, len(lr)) self.assertEqual(rr.n_p, len(lr))
self.assertEqual(True, lr == OGRGeometry(rr.wkt)) self.assertEqual(lr, OGRGeometry(rr.wkt))
self.assertEqual(True, lr != prev) self.assertNotEqual(lr, prev)
prev = lr prev = lr
def test07a_polygons(self): def test07a_polygons(self):
@ -222,8 +223,8 @@ class OGRGeomTest(unittest.TestCase, TestDataMixin):
self.assertAlmostEqual(p.centroid[1], y, 9) self.assertAlmostEqual(p.centroid[1], y, 9)
# Testing equivalence # Testing equivalence
self.assertEqual(True, poly == OGRGeometry(p.wkt)) self.assertEqual(poly, OGRGeometry(p.wkt))
self.assertEqual(True, poly != prev) self.assertNotEqual(poly, prev)
if p.ext_ring_cs: if p.ext_ring_cs:
ring = poly[0] ring = poly[0]
@ -367,7 +368,7 @@ class OGRGeomTest(unittest.TestCase, TestDataMixin):
a = OGRGeometry(self.geometries.topology_geoms[i].wkt_a) a = OGRGeometry(self.geometries.topology_geoms[i].wkt_a)
b = OGRGeometry(self.geometries.topology_geoms[i].wkt_b) b = OGRGeometry(self.geometries.topology_geoms[i].wkt_b)
i1 = OGRGeometry(self.geometries.intersect_geoms[i].wkt) i1 = OGRGeometry(self.geometries.intersect_geoms[i].wkt)
self.assertEqual(True, a.intersects(b)) self.assertTrue(a.intersects(b))
i2 = a.intersection(b) i2 = a.intersection(b)
self.assertEqual(i1, i2) self.assertEqual(i1, i2)
self.assertEqual(i1, a & b) # __and__ is intersection operator self.assertEqual(i1, a & b) # __and__ is intersection operator
@ -478,5 +479,5 @@ class OGRGeomTest(unittest.TestCase, TestDataMixin):
def test19_equivalence_regression(self): def test19_equivalence_regression(self):
"Testing equivalence methods with non-OGRGeometry instances." "Testing equivalence methods with non-OGRGeometry instances."
self.assertNotEqual(None, OGRGeometry('POINT(0 0)')) self.assertIsNotNone(OGRGeometry('POINT(0 0)'))
self.assertEqual(False, OGRGeometry('LINESTRING(0 0, 1 1)') == 3) self.assertNotEqual(OGRGeometry('LINESTRING(0 0, 1 1)'), 3)