You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
333 lines
8.7 KiB
333 lines
8.7 KiB
""" |
|
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()
|
|
|