~canonical-django/canonical-django/project-template

« back to all changes in this revision

Viewing changes to trunk/python-packages/django/contrib/gis/tests/test_gdal_driver.py

  • Committer: Matthew Nuzum
  • Date: 2008-11-13 05:46:03 UTC
  • Revision ID: matthew.nuzum@canonical.com-20081113054603-v0kvr6z6xyexvqt3
adding to version control

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
import os, os.path, unittest
 
2
from django.contrib.gis.gdal import Driver, OGRException
 
3
 
 
4
valid_drivers = ('ESRI Shapefile', 'MapInfo File', 'TIGER', 'S57', 'DGN',
 
5
                 'Memory', 'CSV', 'GML', 'KML')
 
6
 
 
7
invalid_drivers = ('Foo baz', 'clucka', 'ESRI Shp')
 
8
 
 
9
aliases = {'eSrI' : 'ESRI Shapefile',
 
10
           'TigER/linE' : 'TIGER',
 
11
           'SHAPE' : 'ESRI Shapefile',
 
12
           'sHp' : 'ESRI Shapefile',
 
13
           }
 
14
 
 
15
class DriverTest(unittest.TestCase):
 
16
 
 
17
    def test01_valid_driver(self):
 
18
        "Testing valid OGR Data Source Drivers."
 
19
        for d in valid_drivers:
 
20
            dr = Driver(d)
 
21
            self.assertEqual(d, str(dr))
 
22
 
 
23
    def test02_invalid_driver(self):
 
24
        "Testing invalid OGR Data Source Drivers."
 
25
        for i in invalid_drivers:
 
26
            self.assertRaises(OGRException, Driver, i)
 
27
 
 
28
    def test03_aliases(self):
 
29
        "Testing driver aliases."
 
30
        for alias, full_name in aliases.items():
 
31
            dr = Driver(alias)
 
32
            self.assertEqual(full_name, str(dr))
 
33
 
 
34
def suite():
 
35
    s = unittest.TestSuite()
 
36
    s.addTest(unittest.makeSuite(DriverTest))
 
37
    return s
 
38
 
 
39
def run(verbosity=2):
 
40
    unittest.TextTestRunner(verbosity=verbosity).run(suite())