diff -r 6641e941ef1e -r ff1a9aa48cfd app/django/contrib/gis/tests/test_measure.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/app/django/contrib/gis/tests/test_measure.py Tue Oct 14 16:00:59 2008 +0000 @@ -0,0 +1,333 @@ +""" +Distance and Area objects to allow for sensible and convienient calculation +and conversions. Here are some tests. +""" + +import unittest +from django.contrib.gis.measure import Distance, Area, D, A + +class DistanceTest(unittest.TestCase): + "Testing the Distance object" + + def testInit(self): + "Testing initialisation from valid units" + d = Distance(m=100) + self.assertEqual(d.m, 100) + + d1, d2, d3 = D(m=100), D(meter=100), D(metre=100) + for d in (d1, d2, d3): + self.assertEqual(d.m, 100) + + d = D(nm=100) + self.assertEqual(d.m, 185200) + + y1, y2, y3 = D(yd=100), D(yard=100), D(Yard=100) + for d in (y1, y2, y3): + self.assertEqual(d.yd, 100) + + mm1, mm2 = D(millimeter=1000), D(MiLLiMeTeR=1000) + for d in (mm1, mm2): + self.assertEqual(d.m, 1.0) + self.assertEqual(d.mm, 1000.0) + + + def testInitInvalid(self): + "Testing initialisation from invalid units" + self.assertRaises(AttributeError, D, banana=100) + + def testAccess(self): + "Testing access in different units" + d = D(m=100) + self.assertEqual(d.km, 0.1) + self.assertAlmostEqual(d.ft, 328.084, 3) + + def testAccessInvalid(self): + "Testing access in invalid units" + d = D(m=100) + self.failIf(hasattr(d, 'banana')) + + def testAddition(self): + "Test addition & subtraction" + d1 = D(m=100) + d2 = D(m=200) + + d3 = d1 + d2 + self.assertEqual(d3.m, 300) + d3 += d1 + self.assertEqual(d3.m, 400) + + d4 = d1 - d2 + self.assertEqual(d4.m, -100) + d4 -= d1 + self.assertEqual(d4.m, -200) + + try: + d5 = d1 + 1 + except TypeError, e: + pass + else: + self.fail('Distance + number should raise TypeError') + + try: + d5 = d1 - 1 + except TypeError, e: + pass + else: + self.fail('Distance - number should raise TypeError') + + try: + d1 += 1 + except TypeError, e: + pass + else: + self.fail('Distance += number should raise TypeError') + + try: + d1 -= 1 + except TypeError, e: + pass + else: + self.fail('Distance -= number should raise TypeError') + + def testMultiplication(self): + "Test multiplication & division" + d1 = D(m=100) + + d3 = d1 * 2 + self.assertEqual(d3.m, 200) + d3 *= 5 + self.assertEqual(d3.m, 1000) + + d4 = d1 / 2 + self.assertEqual(d4.m, 50) + d4 /= 5 + self.assertEqual(d4.m, 10) + + a5 = d1 * D(m=10) + self.assert_(isinstance(a5, Area)) + self.assertEqual(a5.sq_m, 100*10) + + try: + d1 *= D(m=1) + except TypeError, e: + pass + else: + self.fail('Distance *= Distance should raise TypeError') + + try: + d5 = d1 / D(m=1) + except TypeError, e: + pass + else: + self.fail('Distance / Distance should raise TypeError') + + try: + d1 /= D(m=1) + except TypeError, e: + pass + else: + self.fail('Distance /= Distance should raise TypeError') + + def testUnitConversions(self): + "Testing default units during maths" + d1 = D(m=100) + d2 = D(km=1) + + d3 = d1 + d2 + self.assertEqual(d3._default_unit, 'm') + d4 = d2 + d1 + self.assertEqual(d4._default_unit, 'km') + d5 = d1 * 2 + self.assertEqual(d5._default_unit, 'm') + d6 = d1 / 2 + self.assertEqual(d6._default_unit, 'm') + + def testComparisons(self): + "Testing comparisons" + d1 = D(m=100) + d2 = D(km=1) + d3 = D(km=0) + + self.assert_(d2 > d1) + self.assert_(d1 == d1) + self.assert_(d1 < d2) + self.failIf(d3) + + def testUnitsStr(self): + "Testing conversion to strings" + d1 = D(m=100) + d2 = D(km=3.5) + + self.assertEqual(str(d1), '100.0 m') + self.assertEqual(str(d2), '3.5 km') + self.assertEqual(repr(d1), 'Distance(m=100.0)') + self.assertEqual(repr(d2), 'Distance(km=3.5)') + + def testUnitAttName(self): + "Testing the `unit_attname` class method" + unit_tuple = [('Yard', 'yd'), ('Nautical Mile', 'nm'), ('German legal metre', 'german_m'), + ('Indian yard', 'indian_yd'), ('Chain (Sears)', 'chain_sears'), ('Chain', 'chain')] + for nm, att in unit_tuple: + self.assertEqual(att, D.unit_attname(nm)) + +class AreaTest(unittest.TestCase): + "Testing the Area object" + + def testInit(self): + "Testing initialisation from valid units" + a = Area(sq_m=100) + self.assertEqual(a.sq_m, 100) + + a = A(sq_m=100) + self.assertEqual(a.sq_m, 100) + + a = A(sq_mi=100) + self.assertEqual(a.sq_m, 258998811.0336) + + def testInitInvaliA(self): + "Testing initialisation from invalid units" + self.assertRaises(AttributeError, A, banana=100) + + def testAccess(self): + "Testing access in different units" + a = A(sq_m=100) + self.assertEqual(a.sq_km, 0.0001) + self.assertAlmostEqual(a.sq_ft, 1076.391, 3) + + def testAccessInvaliA(self): + "Testing access in invalid units" + a = A(sq_m=100) + self.failIf(hasattr(a, 'banana')) + + def testAddition(self): + "Test addition & subtraction" + a1 = A(sq_m=100) + a2 = A(sq_m=200) + + a3 = a1 + a2 + self.assertEqual(a3.sq_m, 300) + a3 += a1 + self.assertEqual(a3.sq_m, 400) + + a4 = a1 - a2 + self.assertEqual(a4.sq_m, -100) + a4 -= a1 + self.assertEqual(a4.sq_m, -200) + + try: + a5 = a1 + 1 + except TypeError, e: + pass + else: + self.fail('Area + number should raise TypeError') + + try: + a5 = a1 - 1 + except TypeError, e: + pass + else: + self.fail('Area - number should raise TypeError') + + try: + a1 += 1 + except TypeError, e: + pass + else: + self.fail('Area += number should raise TypeError') + + try: + a1 -= 1 + except TypeError, e: + pass + else: + self.fail('Area -= number should raise TypeError') + + def testMultiplication(self): + "Test multiplication & division" + a1 = A(sq_m=100) + + a3 = a1 * 2 + self.assertEqual(a3.sq_m, 200) + a3 *= 5 + self.assertEqual(a3.sq_m, 1000) + + a4 = a1 / 2 + self.assertEqual(a4.sq_m, 50) + a4 /= 5 + self.assertEqual(a4.sq_m, 10) + + try: + a5 = a1 * A(sq_m=1) + except TypeError, e: + pass + else: + self.fail('Area * Area should raise TypeError') + + try: + a1 *= A(sq_m=1) + except TypeError, e: + pass + else: + self.fail('Area *= Area should raise TypeError') + + try: + a5 = a1 / A(sq_m=1) + except TypeError, e: + pass + else: + self.fail('Area / Area should raise TypeError') + + try: + a1 /= A(sq_m=1) + except TypeError, e: + pass + else: + self.fail('Area /= Area should raise TypeError') + + def testUnitConversions(self): + "Testing default units during maths" + a1 = A(sq_m=100) + a2 = A(sq_km=1) + + a3 = a1 + a2 + self.assertEqual(a3._default_unit, 'sq_m') + a4 = a2 + a1 + self.assertEqual(a4._default_unit, 'sq_km') + a5 = a1 * 2 + self.assertEqual(a5._default_unit, 'sq_m') + a6 = a1 / 2 + self.assertEqual(a6._default_unit, 'sq_m') + + def testComparisons(self): + "Testing comparisons" + a1 = A(sq_m=100) + a2 = A(sq_km=1) + a3 = A(sq_km=0) + + self.assert_(a2 > a1) + self.assert_(a1 == a1) + self.assert_(a1 < a2) + self.failIf(a3) + + def testUnitsStr(self): + "Testing conversion to strings" + a1 = A(sq_m=100) + a2 = A(sq_km=3.5) + + self.assertEqual(str(a1), '100.0 sq_m') + self.assertEqual(str(a2), '3.5 sq_km') + self.assertEqual(repr(a1), 'Area(sq_m=100.0)') + self.assertEqual(repr(a2), 'Area(sq_km=3.5)') + + +def suite(): + s = unittest.TestSuite() + s.addTest(unittest.makeSuite(DistanceTest)) + s.addTest(unittest.makeSuite(AreaTest)) + return s + +def run(verbosity=2): + unittest.TextTestRunner(verbosity=verbosity).run(suite()) + +if __name__=="__main__": + run()