test_driver.py 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. import unittest
  2. from unittest import skipUnless
  3. from django.contrib.gis.gdal import HAS_GDAL
  4. if HAS_GDAL:
  5. from django.contrib.gis.gdal import Driver, OGRException
  6. valid_drivers = ('ESRI Shapefile', 'MapInfo File', 'TIGER', 'S57', 'DGN',
  7. 'Memory', 'CSV', 'GML', 'KML')
  8. invalid_drivers = ('Foo baz', 'clucka', 'ESRI Shp')
  9. aliases = {'eSrI': 'ESRI Shapefile',
  10. 'TigER/linE': 'TIGER',
  11. 'SHAPE': 'ESRI Shapefile',
  12. 'sHp': 'ESRI Shapefile',
  13. }
  14. @skipUnless(HAS_GDAL, "GDAL is required")
  15. class DriverTest(unittest.TestCase):
  16. def test01_valid_driver(self):
  17. "Testing valid OGR Data Source Drivers."
  18. for d in valid_drivers:
  19. dr = Driver(d)
  20. self.assertEqual(d, str(dr))
  21. def test02_invalid_driver(self):
  22. "Testing invalid OGR Data Source Drivers."
  23. for i in invalid_drivers:
  24. self.assertRaises(OGRException, Driver, i)
  25. def test03_aliases(self):
  26. "Testing driver aliases."
  27. for alias, full_name in aliases.items():
  28. dr = Driver(alias)
  29. self.assertEqual(full_name, str(dr))