/[thuban]/branches/WIP-pyshapelib-bramz/test/test_proj.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/test/test_proj.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1813 by bh, Mon Oct 13 14:01:37 2003 UTC revision 1933 by bh, Tue Nov 11 16:37:53 2003 UTC
# Line 23  support.initthuban() Line 23  support.initthuban()
23  from Thuban import _  from Thuban import _
24  from Thuban.Model.proj import Projection, ProjFile, \  from Thuban.Model.proj import Projection, ProjFile, \
25       PROJ_UNITS_METERS, PROJ_UNITS_DEGREES       PROJ_UNITS_METERS, PROJ_UNITS_DEGREES
26    from Thuban.Model.messages import PROJECTION_ADDED, PROJECTION_REMOVED, \
27         PROJECTION_REPLACED
28  import Thuban.Model.resource as resource  import Thuban.Model.resource as resource
29    
30  from xmlsupport import sax_eventlist  from xmlsupport import sax_eventlist
# Line 79  class TestProjection(unittest.TestCase, Line 80  class TestProjection(unittest.TestCase,
80          self.assertEquals(proj.GetParameter("south"), "south")          self.assertEquals(proj.GetParameter("south"), "south")
81    
82      def test_get_projection_units_geo(self):      def test_get_projection_units_geo(self):
83          """Test Projection.GetProjectedUnits() for geographic projection"""          """Test Projection.GetProjectedUnits() for geographic projection.
84            Test for the alias 'longlat' as well.
85            """
86          proj = Projection(["proj=latlong", "to_meter=0.017453292519943295",          proj = Projection(["proj=latlong", "to_meter=0.017453292519943295",
87                             "ellps=clrk66"])                             "ellps=clrk66"])
88          self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_DEGREES)          self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_DEGREES)
89            proj = Projection(["proj=longlat", "to_meter=0.017453292519943295",
90                               "ellps=clrk66"])
91            self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_DEGREES)
92    
93      def test_get_projection_units_normal(self):      def test_get_projection_units_normal(self):
94          """Test Projection.GetProjectedUnits() for normal projection"""          """Test Projection.GetProjectedUnits() for normal projection"""
95          proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])          proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])
96          self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_METERS)          self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_METERS)
97    
98        def test_label(self):
99            """Test Projection.Label() without epsg"""
100            proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"],
101                               name = "My Projection")
102            self.assertEquals(proj.Label(), "My Projection")
103    
104        def test_label_epsg(self):
105            """Test Projection.Label() with epsg"""
106            proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"],
107                               name = "My Projection", epsg="42")
108            self.assertEquals(proj.Label(), "EPSG    42 My Projection")
109    
110        def test_epsgcode_for_non_epsg_projection(self):
111            """Test Projection.EPSGCode() without epsg"""
112            proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"],
113                               name = "My Projection")
114            self.assertEquals(proj.EPSGCode(), None)
115    
116        def test_epsgcode_for_real_epsg_projection(self):
117            """Test Projection.EPSGCode() with epsg"""
118            proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"],
119                               name = "My Projection", epsg="42")
120            self.assertEquals(proj.EPSGCode(), "42")
121    
 sample_projfile = '''\  
 <?xml version="1.0" encoding="UTF-8"?>  
 <!DOCTYPE projectionlist SYSTEM "projfile.dtd">  
 <projectionlist>  
     <projection name="Transverse Mercator">  
         <parameter value="proj=tmerc"/>  
         <parameter value="ellps=clrk66"/>  
         <parameter value="lat_0=90w"/>  
         <parameter value="lon_0=90w"/>  
         <parameter value="k=1"/>  
     </projection>  
     <projection name="Transverse Mercator">  
         <parameter value="proj=tmerc"/>  
         <parameter value="ellps=clrk66"/>  
         <parameter value="lat_0=30w"/>  
         <parameter value="lon_0=30w"/>  
         <parameter value="k=1"/>  
     </projection>  
     <projection name="Universal Transverse Mercator">  
         <parameter value="proj=utm"/>  
         <parameter value="ellps=clrk66"/>  
         <parameter value="zone=1"/>  
     </projection>  
 </projectionlist>  
 '''  
122    
 sample_projfile_data = [("Transverse Mercator", ["proj=tmerc",  
                                                   "ellps=clrk66",  
                                                   "lat_0=90w",  
                                                   "lon_0=90w",  
                                                   "k=1"]),  
                         ("Transverse Mercator", ["proj=tmerc",  
                                                   "ellps=clrk66",  
                                                   "lat_0=30w",  
                                                   "lon_0=30w",  
                                                   "k=1"]),  
                         ("Universal Transverse Mercator", ["proj=utm",  
                                                             "ellps=clrk66",  
                                                             "zone=1"])]  
123    
124  sample_projfile2 = '''\  class TestProjFileSimple:
125  <?xml version="1.0" encoding="UTF-8"?>  
126  <!DOCTYPE projectionlist SYSTEM "projfile.dtd">      def test_init(self):
127  <projectionlist>          """Test ProjFile coinstructor"""
128  </projectionlist>          proj_file = ProjFile("some_filename")
129  '''          self.assertEquals(proj_file.GetFilename(), "some_filename")
130            self.assertEquals(len(proj_file.GetProjections()), 0)
131    
132        def test_set_filename(self):
133            """Test ProjFile.SetFilename()"""
134            proj_file = ProjFile("some_filename")
135            proj.SetFilename("other_name")
136            self.assertEquals(proj_file.GetFilename(), "other_name")
137    
 sample_projfile_data2 = []  
138    
139  class ProjFileTest(unittest.TestCase, support.FileTestMixin):  class TestProjFile(unittest.TestCase, support.SubscriberMixin):
140    
141      """Base class for the proj file tests"""      """Test cases for ProjFile objects"""
142    
143      def filename(self):      def setUp(self):
144          """Return the filename for the test"""          self.clear_messages()
145          return self.temp_file_name(self.id() + ".proj")          self.proj0 = Projection(["proj=tmerc", "ellps=clrk66"])
146            self.proj1 = Projection(["proj=utm", "ellps=clrk66"])
147            self.proj2 = Projection(["proj=lcc", "ellps=clrk66",
148                                     "lat_1=0", "lat_2=20"])
149            self.proj_file = ProjFile("some_filename")
150            for msg in [PROJECTION_ADDED, PROJECTION_REMOVED, PROJECTION_REPLACED]:
151                self.proj_file.Subscribe(msg, self.subscribe_with_params, msg)
152    
153        def tearDown(self):
154            self.clear_messages()
155            self.proj_file.Destroy()
156    
157        def test_add_remove(self):
158            """Test ProjFile.Add() and ProjFile.Remove()"""
159            self.proj_file.Add(self.proj0)
160            self.proj_file.Add(self.proj1)
161            self.assertEquals(self.proj_file.GetProjections(),
162                              [self.proj0, self.proj1])
163            self.check_messages([(self.proj0, PROJECTION_ADDED),
164                                 (self.proj1, PROJECTION_ADDED)])
165            self.clear_messages()
166    
167            self.proj_file.Remove(self.proj0)
168            self.assertEquals(self.proj_file.GetProjections(), [self.proj1])
169            self.check_messages([(self.proj0, PROJECTION_REMOVED)])
170    
171        def test_remove_non_existing(self):
172            """Test ProjFile.Remove(<proj not in projfile>)"""
173            self.assertRaises(ValueError, self.proj_file.Remove, self.proj0)
174            # Nothing happened, so no messages should have been sent
175            self.check_messages([])
176    
177        def test_replace(self):
178            """Test ProjFile.Replace()"""
179            self.proj_file.Add(self.proj0)
180            self.proj_file.Add(self.proj1)
181            self.clear_messages()
182    
183            # Replace()
184            self.proj_file.Replace(self.proj0, self.proj2)
185            self.assertEquals(self.proj_file.GetProjections(),
186                              [self.proj2, self.proj1])
187            self.check_messages([(self.proj0, self.proj2, PROJECTION_REPLACED)])
188    
189        def test_replace_non_existing(self):
190            """Test ProjFile.Replace(<proj not in projfile>, <some proj>)"""
191            self.proj_file.Add(self.proj0)
192            self.proj_file.Add(self.proj1)
193            self.clear_messages()
194            self.assertRaises(ValueError,
195                              self.proj_file.Replace, self.proj2, self.proj0)
196            # All projections should still be there
197            self.assertEquals(self.proj_file.GetProjections(),
198                              [self.proj0, self.proj1])
199            # Nothing happened, so no messages should have been sent
200            self.check_messages([])
201    
 class TestProjFile(ProjFileTest, xmlsupport.ValidationTest):  
202    
203      """Test cases for reading and writing projection files.  class ProjFileTest(unittest.TestCase, support.FileTestMixin,
204      """                     xmlsupport.ValidationTest):
205    
206      def test(self):      """Base class for the proj file tests that read or write files"""
207          """Test ProjFile"""  
208        def filename(self):
209            """Return the filename for the test"""
210            return self.temp_file_name(self.id() + ".proj")
211    
         proj0 = Projection(["proj=tmerc", "ellps=clrk66"])  
         proj1 = Projection(["proj=utm", "ellps=clrk66"])  
         proj2 = Projection(["proj=lcc", "ellps=clrk66",  
                             "lat_1=0", "lat_2=20"])  
   
         eq = self.assertEquals  
   
   
         #  
         # __init__()  
         # GetFilename()  
         # SetFilename()  
         #  
         for name in ["", "hello_world"]:  
             projFile = ProjFile(name)  
             eq(projFile.GetFilename(), name)  
   
             projFile.SetFilename("XXX")  
             projFile.SetFilename(name)  
             eq(projFile.GetFilename(), name)  
   
         # initial number of projections should be 0  
         eq(len(projFile.GetProjections()), 0)  
   
         #  
         # Add()  
         # Remove()  
         #  
         projFile.Add(proj0)  
         eq(len(projFile.GetProjections()), 1)  
         projFile.Remove(proj0)  
         eq(len(projFile.GetProjections()), 0)  
   
         # try to remove something that doesn't exist  
         self.assertRaises(ValueError, projFile.Remove, None)  
         self.assertRaises(ValueError, projFile.Remove, proj0)  
   
         projFile.Add(proj0)  
         projFile.Add(proj1)  
         projFile.Add(proj2)  
         eq(len(projFile.GetProjections()), 3)  
   
         # GetProjections() -- tests order  
         projs = projFile.GetProjections()  
         eq(projs[0], proj0)  
         eq(projs[1], proj1)  
         eq(projs[2], proj2)  
212    
213          projFile.Remove(proj2)  class ProjFileReadTests(ProjFileTest):
         projFile.Remove(proj1)  
214    
215          # Replace()      """Test read ProjFile objects from files
216          projFile.Replace(proj0, proj1)  
217          projs = projFile.GetProjections()      The files only cover error handling and the system projection file.
218          eq(projs[0], proj1)      """
   
         # replace a non-existent projection  
         self.assertRaises(ValueError, projFile.Replace, None, proj2)  
         self.assertRaises(ValueError, projFile.Replace, proj0, proj2)  
   
     def testRead(self):  
         """Test read_proj_file"""  
         self.doTestRead(sample_projfile_data, sample_projfile)  
         self.doTestRead(sample_projfile_data2, sample_projfile2)  
219    
220      def test_read_non_existing_file(self):      def test_read_non_existing_file(self):
221          """Test read_proj_file with non-existing file"""          """Test read_proj_file with non-existing file"""
# Line 238  class TestProjFile(ProjFileTest, xmlsupp Line 234  class TestProjFile(ProjFileTest, xmlsupp
234          file.close()          file.close()
235          os.chmod(filename, 0200) # write-only          os.chmod(filename, 0200) # write-only
236          self.assertRaises(IOError, resource.read_proj_file, filename)          self.assertRaises(IOError, resource.read_proj_file, filename)
         os.chmod(filename, 0600) # read/write so we reuse the file  
237    
238      def test_read_empty_file(self):      def test_read_empty_file(self):
239          """Test read_proj_file with empty file"""          """Test read_proj_file with empty file"""
# Line 248  class TestProjFile(ProjFileTest, xmlsupp Line 243  class TestProjFile(ProjFileTest, xmlsupp
243    
244          self.assertRaises(SAXParseException, resource.read_proj_file, filename)          self.assertRaises(SAXParseException, resource.read_proj_file, filename)
245    
     def doTestRead(self, data, input):  
   
         filename = self.filename()  
         file = open(filename, "w")  
         file.write(input)  
         file.close()  
   
         pf, warnings = resource.read_proj_file(filename)  
         self.assertEquals(warnings, [])  
   
         eq = self.assertEquals  
   
         eq(pf.GetFilename(), filename)  
   
         for proj, d in zip(pf.GetProjections(), data):  
             eq(proj.GetName(), d[0])  
             for param in proj.GetAllParameters():  
                 self.assert_(param in d[1])  
   
246      def test_get_system_proj_file(self):      def test_get_system_proj_file(self):
247          """Test resource.get_system_proj_file()          """Test resource.get_system_proj_file(DEFAULT_PROJ_FILE)
248    
249          This is primarily to test whether the system proj file contains          This is primarily to test whether the system proj file contains
250          invalid projection paramers and whether the proj file is not          invalid projection paramers and whether the proj file is not
251          empty          empty
252          """          """
253          projfile, warnings = resource.get_system_proj_file()          projfile, warnings\
254                      = resource.get_system_proj_file(resource.DEFAULT_PROJ_FILE)
255          self.assertEquals(warnings, [])          self.assertEquals(warnings, [])
256          self.assert_(len(projfile.GetProjections()) > 0)          self.assert_(len(projfile.GetProjections()) > 0)
257    
258            # see whether it got cached and we get the same projfile object
259            # when we read the file again
260            projfile2, warnings \
261                       = resource.get_system_proj_file(resource.DEFAULT_PROJ_FILE)
262            self.assert_(projfile is projfile2)
263    
264    
265  class WriteProjFileTests(ProjFileTest, xmlsupport.ValidationTest):  class WriteProjFileTests(ProjFileTest):
266    
267      """Test cases for writing proj files"""      """Test cases for writing proj files"""
268    
269      def compare_xml(self, xml1, xml2):      def compare_xml(self, xml1, xml2):
270          self.assertEquals(sax_eventlist(xml1), sax_eventlist(xml2))          self.assertEquals(sax_eventlist(xml1), sax_eventlist(xml2))
271    
272      def doTestWrite(self, data, expected):      def doTestWrite(self, projfile, expected):
273          filename = self.filename()          filename = self.filename()
274    
275          pf = ProjFile(filename)          resource.write_proj_file(projfile)
         for proj in data:  
             pf.Add(Projection(proj[1], proj[0]))  
   
         resource.write_proj_file(pf)  
276    
277          file = open(filename)          file = open(filename)
278          written_contents = file.read()          written_contents = file.read()
# Line 304  class WriteProjFileTests(ProjFileTest, x Line 283  class WriteProjFileTests(ProjFileTest, x
283    
284      def test_write(self):      def test_write(self):
285          """Test write_proj_file"""          """Test write_proj_file"""
286          self.doTestWrite(sample_projfile_data, sample_projfile)          pf = ProjFile(self.filename())
287            pf.Add(Projection(['proj=tmerc', 'ellps=clrk66',
288                               'lat_0=90w', 'lon_0=90w', 'k=1'],
289                              "Transverse Mercator",))
290            pf.Add(Projection(["proj=tmerc",
291                               "lat_0=0.000000000", "lon_0=-62.000000000",
292                               "k=0.999500", "x_0=400000.000", "y_0=0.000",
293                               "ellps=clrk80", "units=m"],
294                              "Anguilla 1957 / British West Indies Grid",
295                              epsg="200"))
296            file_contents = '''<?xml version="1.0" encoding="UTF-8"?>
297            <!DOCTYPE projectionlist SYSTEM "projfile.dtd">
298            <projectionlist>
299                <projection name="Transverse Mercator">
300                    <parameter value="proj=tmerc"/>
301                    <parameter value="ellps=clrk66"/>
302                    <parameter value="lat_0=90w"/>
303                    <parameter value="lon_0=90w"/>
304                    <parameter value="k=1"/>
305                </projection>
306                <projection epsg="200"
307                            name="Anguilla 1957 / British West Indies Grid">
308                    <parameter value="proj=tmerc"/>
309                    <parameter value="lat_0=0.000000000"/>
310                    <parameter value="lon_0=-62.000000000"/>
311                    <parameter value="k=0.999500"/>
312                    <parameter value="x_0=400000.000"/>
313                    <parameter value="y_0=0.000"/>
314                    <parameter value="ellps=clrk80"/>
315                    <parameter value="units=m"/>
316                </projection>
317            </projectionlist>
318            '''
319            self.doTestWrite(pf, file_contents)
320    
321      def test_write_empty_file(self):      def test_write_empty_file(self):
322          """Test write empty ProjFile"""          """Test write empty ProjFile"""
323          self.doTestWrite(sample_projfile_data2, sample_projfile2)          pf = ProjFile(self.filename())
324            file_contents = '''<?xml version="1.0" encoding="UTF-8"?>
325            <!DOCTYPE projectionlist SYSTEM "projfile.dtd">
326            <projectionlist>
327            </projectionlist>
328            '''
329            self.doTestWrite(pf, file_contents)
330    
331    
332    class ProjFileLoadTestCase(support.FileLoadTestCase):
333    
334  class TestProjFileWithInvalidParameters(unittest.TestCase,      """Base class for the test cases that read specific test files"""
                                         support.FileLoadTestCase):  
335    
336      file_extension = ".proj"      file_extension = ".proj"
337    
338        def tearDown(self):
339            """Clear the cache explicitly"""
340            resource.clear_proj_file_cache()
341    
342    
343    class TestLoadingProjFile(ProjFileLoadTestCase):
344    
345        file_contents = '''\
346    <?xml version="1.0" encoding="UTF-8"?>
347    <!DOCTYPE projectionlist SYSTEM "projfile.dtd">
348    <projectionlist>
349        <projection name="Transverse Mercator">
350            <parameter value="proj=tmerc"/>
351            <parameter value="ellps=clrk66"/>
352            <parameter value="lat_0=90w"/>
353            <parameter value="lon_0=90w"/>
354            <parameter value="k=1"/>
355        </projection>
356        <projection epsg="200" name="Anguilla 1957 / British West Indies Grid">
357            <parameter value="proj=tmerc"/>
358            <parameter value="lat_0=0.000000000"/>
359            <parameter value="lon_0=-62.000000000"/>
360            <parameter value="k=0.999500"/>
361            <parameter value="x_0=400000.000"/>
362            <parameter value="y_0=0.000"/>
363            <parameter value="ellps=clrk80"/>
364            <parameter value="units=m"/>
365        </projection>
366    </projectionlist>
367    '''
368    
369        def check_projection(self, proj, label, parameters):
370            """Check the values of the proj's label and parameters"""
371            self.assertEquals(proj.Label(), label)
372            params = proj.GetAllParameters()[:]
373            params.sort()
374            self.assertEquals(params, parameters)
375    
376        def test(self):
377            projfile, warnings = resource.read_proj_file(self.filename())
378            # no warnings
379            self.assertEquals(warnings, [])
380    
381            # There are two projections
382            projs = projfile.GetProjections()
383            self.assertEquals(len(projs), 2)
384    
385            self.check_projection(projs[0],
386                                  "Transverse Mercator",
387                                  ['ellps=clrk66', 'k=1', 'lat_0=90w', 'lon_0=90w',
388                                   'proj=tmerc'])
389            self.check_projection(projs[1],
390                             "EPSG   200 Anguilla 1957 / British West Indies Grid",
391                                  ["ellps=clrk80", "k=0.999500",
392                                   "lat_0=0.000000000", "lon_0=-62.000000000",
393                                   "proj=tmerc", "units=m",
394                                   "x_0=400000.000", "y_0=0.000"])
395    
396        def test_caching(self):
397            # test whether the projfile cache works
398            projfile, warnings = resource.read_proj_file(self.filename())
399            projfile2, warnings = resource.read_proj_file(self.filename())
400    
401            # Both projfiles should be the same object
402            self.assert_(projfile2 is projfile)
403    
404            # If we clear the cache we should get a new one.
405            resource.clear_proj_file_cache()
406            projfile3, warnings = resource.read_proj_file(self.filename())
407            self.assert_(projfile3 is not projfile)
408    
409    
410    class TestLoadingProjFileWithEmptyProjectionlist(ProjFileLoadTestCase):
411    
412        file_contents = '''\
413    <?xml version="1.0" encoding="UTF-8"?>
414    <!DOCTYPE projectionlist SYSTEM "projfile.dtd">
415    <projectionlist>
416    </projectionlist>
417    '''
418    
419        def test(self):
420            projfile, warnings = resource.read_proj_file(self.filename())
421            # no warnings
422            self.assertEquals(warnings, [])
423    
424            # There are no projections
425            self.assertEquals(len(projfile.GetProjections()), 0)
426    
427    
428    class TestProjFileWithInvalidParameters(ProjFileLoadTestCase):
429    
430      file_contents = '''\      file_contents = '''\
431  <?xml version="1.0" encoding="UTF-8"?>  <?xml version="1.0" encoding="UTF-8"?>
432  <!DOCTYPE projectionlist SYSTEM "projfile.dtd">  <!DOCTYPE projectionlist SYSTEM "projfile.dtd">
# Line 354  class TestProjFileWithInvalidParameters( Line 466  class TestProjFileWithInvalidParameters(
466    
467    
468  if __name__ == "__main__":  if __name__ == "__main__":
469      unittest.main()      support.run_tests()
Legend:
Removed from v.1813  
changed lines
  Added in v.1933
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26