/[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 752 by jonathan, Fri Apr 25 14:23:34 2003 UTC revision 1798 by bh, Thu Oct 9 10:17:49 2003 UTC
# Line 1  Line 1 
1  # Copyright (c) 2002 by Intevation GmbH  # Copyright (c) 2002, 2003 by Intevation GmbH
2  # Authors:  # Authors:
3  # Bernhard Herzog <[email protected]>  # Bernhard Herzog <[email protected]>
4  #  #
# Line 16  __version__ = "$Revision$" Line 16  __version__ = "$Revision$"
16  import unittest  import unittest
17  import os  import os
18    
19    import xmlsupport
20  import support  import support
21  support.initthuban()  support.initthuban()
22    
23  from Thuban.Model.proj import Projection, ProjFile  from Thuban import _
24    from Thuban.Model.proj import Projection, ProjFile, \
25         PROJ_UNITS_METERS, PROJ_UNITS_DEGREES
26    
27  import Thuban.Model.resource as resource  import Thuban.Model.resource as resource
28    
29  from test_save import sax_eventlist  from xmlsupport import sax_eventlist
30    
31  from xml.sax import SAXParseException  from xml.sax import SAXParseException
32    
# Line 35  class TestProjection(unittest.TestCase, Line 38  class TestProjection(unittest.TestCase,
38    
39      def test(self):      def test(self):
40          """Test Projection"""          """Test Projection"""
41          proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])          params = ["zone=26", "proj=utm", "ellps=clrk66"]
42          self.assertEquals(proj.params, ["zone=26", "proj=utm", "ellps=clrk66"])          proj = Projection(params)
43            self.assertEquals(proj.params, params)
44    
45          # It's not clear whether this value is really the correct one          # It's not clear whether this value is really the correct one
46          # but a test failure here probably still means a bug somewhere          # but a test failure here probably still means a bug somewhere
# Line 51  class TestProjection(unittest.TestCase, Line 55  class TestProjection(unittest.TestCase,
55                                    3875381.8535437919, 252962.10480170773),                                    3875381.8535437919, 252962.10480170773),
56                                   epsilon = 1e-5)                                   epsilon = 1e-5)
57    
58            # GetName()
59            self.assertEquals(proj.GetName(), _("Unknown"))
60    
61            # GetParameter()
62          self.assertEquals(proj.GetParameter("zone"), "26")          self.assertEquals(proj.GetParameter("zone"), "26")
63          self.assertEquals(proj.GetParameter("proj"), "utm")          self.assertEquals(proj.GetParameter("proj"), "utm")
64          self.assertEquals(proj.GetParameter("ellps"), "clrk66")          self.assertEquals(proj.GetParameter("ellps"), "clrk66")
65          self.assertEquals(proj.GetParameter("hallo"), "")          self.assertEquals(proj.GetParameter("hallo"), "")
66    
67            # GetAllParameters()
68            self.assertEquals(proj.GetAllParameters(), params)
69    
70            # GetName()
71            proj = Projection(params, "MyName")
72            self.assertEquals(proj.GetName(), "MyName")
73    
74        def test_get_parameter_without_equals_sign(self):
75            """Test Projection.GetParameter() for a parameter without '=' sign"""
76            proj = Projection(["proj=utm", "zone=34", "south", "ellps=clrk66"])
77            # The Projection class pretends that for parameters specified
78            # without a value the value is the same as the parameter name.
79            self.assertEquals(proj.GetParameter("south"), "south")
80    
81        def test_get_projection_units_geo(self):
82            """Test Projection.GetProjectedUnits() for geographic projection"""
83            proj = Projection(["proj=latlong", "to_meter=0.017453292519943295",
84                               "ellps=clrk66"])
85            self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_DEGREES)
86    
87        def test_get_projection_units_normal(self):
88            """Test Projection.GetProjectedUnits() for normal projection"""
89            proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])
90            self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_METERS)
91    
92    
93  sample_projfile = '''\  sample_projfile = '''\
94  <?xml version="1.0" encoding="UTF-8"?>  <?xml version="1.0" encoding="UTF-8"?>
95  <!DOCTYPE projfile SYSTEM "thuban.dtd">  <!DOCTYPE projectionlist SYSTEM "projfile.dtd">
96  <projectionlist>  <projectionlist>
97      <projection name="Transverse Mercartor">      <projection name="Transverse Mercator">
98          <parameter value="proj=tmerc"/>          <parameter value="proj=tmerc"/>
99          <parameter value="ellps=clrk66"/>          <parameter value="ellps=clrk66"/>
100          <parameter value="lat_0=90w"/>          <parameter value="lat_0=90w"/>
101          <parameter value="lon_0=90w"/>          <parameter value="lon_0=90w"/>
102          <parameter value="k=1"/>          <parameter value="k=1"/>
103      </projection>      </projection>
104      <projection name="Transverse Mercartor">      <projection name="Transverse Mercator">
105          <parameter value="proj=tmerc"/>          <parameter value="proj=tmerc"/>
106          <parameter value="ellps=clrk66"/>          <parameter value="ellps=clrk66"/>
107          <parameter value="lat_0=30w"/>          <parameter value="lat_0=30w"/>
108          <parameter value="lon_0=30w"/>          <parameter value="lon_0=30w"/>
109          <parameter value="k=1"/>          <parameter value="k=1"/>
110      </projection>      </projection>
111      <projection name="Universal Transverse Mercartor">      <projection name="Universal Transverse Mercator">
112          <parameter value="proj=utm"/>          <parameter value="proj=utm"/>
113          <parameter value="ellps=clrk66"/>          <parameter value="ellps=clrk66"/>
114          <parameter value="zone=1"/>          <parameter value="zone=1"/>
# Line 83  sample_projfile = '''\ Line 116  sample_projfile = '''\
116  </projectionlist>  </projectionlist>
117  '''  '''
118    
119  sample_projfile_data = [("Transverse Mercartor", ["proj=tmerc",  sample_projfile_data = [("Transverse Mercator", ["proj=tmerc",
120                                                    "ellps=clrk66",                                                    "ellps=clrk66",
121                                                    "lat_0=90w",                                                    "lat_0=90w",
122                                                    "lon_0=90w",                                                    "lon_0=90w",
123                                                    "k=1"]),                                                    "k=1"]),
124                          ("Transverse Mercartor", ["proj=tmerc",                          ("Transverse Mercator", ["proj=tmerc",
125                                                    "ellps=clrk66",                                                    "ellps=clrk66",
126                                                    "lat_0=30w",                                                    "lat_0=30w",
127                                                    "lon_0=30w",                                                    "lon_0=30w",
128                                                    "k=1"]),                                                    "k=1"]),
129                          ("Universal Transverse Mercartor", ["proj=utm",                          ("Universal Transverse Mercator", ["proj=utm",
130                                                              "ellps=clrk66",                                                              "ellps=clrk66",
131                                                              "zone=1"])]                                                              "zone=1"])]
132    
133  sample_projfile2 = '''\  sample_projfile2 = '''\
134  <?xml version="1.0" encoding="UTF-8"?>  <?xml version="1.0" encoding="UTF-8"?>
135  <!DOCTYPE projfile SYSTEM "thuban.dtd">  <!DOCTYPE projectionlist SYSTEM "projfile.dtd">
136  <projectionlist>  <projectionlist>
137  </projectionlist>  </projectionlist>
138  '''  '''
139    
140  sample_projfile_data2 = []  sample_projfile_data2 = []
141    
142  class TestProjFile(unittest.TestCase, support.FileTestMixin):  class TestProjFile(unittest.TestCase, support.FileTestMixin,
143                       xmlsupport.ValidationTest):
144    
145      """Test cases for reading and writing projection files.      """Test cases for reading and writing projection files.
146      """      """
# Line 119  class TestProjFile(unittest.TestCase, su Line 153  class TestProjFile(unittest.TestCase, su
153    
154          proj0 = Projection(["proj=tmerc", "ellps=clrk66"])          proj0 = Projection(["proj=tmerc", "ellps=clrk66"])
155          proj1 = Projection(["proj=utm", "ellps=clrk66"])          proj1 = Projection(["proj=utm", "ellps=clrk66"])
156          proj2 = Projection(["proj=lcc", "ellps=clrk66"])          proj2 = Projection(["proj=lcc", "ellps=clrk66",
157                                "lat_1=0", "lat_2=20"])
158    
159          eq = self.assertEquals          eq = self.assertEquals
160    
161    
162          #          #
163          # __init__()          # __init__()
164          # GetFilename()          # GetFilename()
# Line 149  class TestProjFile(unittest.TestCase, su Line 185  class TestProjFile(unittest.TestCase, su
185          eq(len(projFile.GetProjections()), 0)          eq(len(projFile.GetProjections()), 0)
186    
187          # try to remove something that doesn't exist          # try to remove something that doesn't exist
188            self.assertRaises(ValueError, projFile.Remove, None)
189          self.assertRaises(ValueError, projFile.Remove, proj0)          self.assertRaises(ValueError, projFile.Remove, proj0)
190    
191          projFile.Add(proj0)          projFile.Add(proj0)
# Line 162  class TestProjFile(unittest.TestCase, su Line 199  class TestProjFile(unittest.TestCase, su
199          eq(projs[1], proj1)          eq(projs[1], proj1)
200          eq(projs[2], proj2)          eq(projs[2], proj2)
201    
202          #          projFile.Remove(proj2)
203          # SetAllParameters()          projFile.Remove(proj1)
204          # SetName()  
205          # SetProjection()          # Replace()
206          #          projFile.Replace(proj0, proj1)
207          params = ["proj=utm", "ellps=clrk66", "zone=13"]          projs = projFile.GetProjections()
208          proj0.SetAllParameters(params)          eq(projs[0], proj1)
209          eq(proj0.GetAllParameters(), params)  
210                    # replace a non-existent projection
211          proj2.SetName("something")          self.assertRaises(ValueError, projFile.Replace, None, proj2)
212          proj0.SetProjection(proj2)          self.assertRaises(ValueError, projFile.Replace, proj0, proj2)
         eq(proj0.GetName(), "something")  
         eq(proj0.GetAllParameters(), proj2.GetAllParameters())  
213    
214      def testRead(self):      def testRead(self):
215          """Test ReadProjFile"""          """Test read_proj_file"""
216    
217          self.doTestRead(sample_projfile_data, sample_projfile)          self.doTestRead(sample_projfile_data, sample_projfile)
218          self.doTestRead(sample_projfile_data2, sample_projfile2)          self.doTestRead(sample_projfile_data2, sample_projfile2)
# Line 186  class TestProjFile(unittest.TestCase, su Line 221  class TestProjFile(unittest.TestCase, su
221          # file doesn't exist          # file doesn't exist
222          #          #
223          self.assertRaises(IOError,          self.assertRaises(IOError,
224              resource.ReadProjFile, self.temp_file_name("nonexistent.proj"))              resource.read_proj_file, self.temp_file_name("nonexistent.proj"))
225    
226          #          #
227          # file isn't readable          # file isn't readable
# Line 195  class TestProjFile(unittest.TestCase, su Line 230  class TestProjFile(unittest.TestCase, su
230          file = open(filename, "w")          file = open(filename, "w")
231          file.close()          file.close()
232          os.chmod(filename, 0200) # write-only          os.chmod(filename, 0200) # write-only
233          self.assertRaises(IOError, resource.ReadProjFile, filename)          self.assertRaises(IOError, resource.read_proj_file, filename)
234          os.chmod(filename, 0600) # read/write so we reuse the file          os.chmod(filename, 0600) # read/write so we reuse the file
235    
236          #          #
# Line 205  class TestProjFile(unittest.TestCase, su Line 240  class TestProjFile(unittest.TestCase, su
240          file = open(filename, "w")          file = open(filename, "w")
241          file.close()          file.close()
242    
243          self.assertRaises(SAXParseException, resource.ReadProjFile, filename)          self.assertRaises(SAXParseException, resource.read_proj_file, filename)
244    
245      def testWrite(self):      def testWrite(self):
246          """Test WriteProjFile"""          """Test write_proj_file"""
247    
248          self.doTestWrite(sample_projfile_data, sample_projfile)          self.doTestWrite(sample_projfile_data, sample_projfile)
249          self.doTestWrite(sample_projfile_data2, sample_projfile2)          self.doTestWrite(sample_projfile_data2, sample_projfile2)
# Line 221  class TestProjFile(unittest.TestCase, su Line 256  class TestProjFile(unittest.TestCase, su
256          for proj in data:          for proj in data:
257              pf.Add(Projection(proj[1], proj[0]))              pf.Add(Projection(proj[1], proj[0]))
258    
259          resource.WriteProjFile(pf)          resource.write_proj_file(pf)
260    
261          file = open(filename)          file = open(filename)
262          written_contents = file.read()          written_contents = file.read()
263          file.close()          file.close()
264          self.compare_xml(written_contents, expected)          self.compare_xml(written_contents, expected)
265            self.validate_data(written_contents)
266            self.validate_data(expected)
267    
268      def doTestRead(self, data, input):      def doTestRead(self, data, input):
269    
# Line 235  class TestProjFile(unittest.TestCase, su Line 272  class TestProjFile(unittest.TestCase, su
272          file.write(input)          file.write(input)
273          file.close()          file.close()
274    
275          pf = resource.ReadProjFile(filename)          pf, warnings = resource.read_proj_file(filename)
276            self.assertEquals(warnings, [])
277    
278          eq = self.assertEquals          eq = self.assertEquals
279    
# Line 246  class TestProjFile(unittest.TestCase, su Line 284  class TestProjFile(unittest.TestCase, su
284              for param in proj.GetAllParameters():              for param in proj.GetAllParameters():
285                  self.assert_(param in d[1])                  self.assert_(param in d[1])
286    
287        def test_get_system_proj_file(self):
288            """Test resource.get_system_proj_file()
289    
290            This is primarily to test whether the system proj file contains
291            invalid projection paramers and whether the proj file is not
292            empty
293            """
294            projfile, warnings = resource.get_system_proj_file()
295            self.assertEquals(warnings, [])
296            self.assert_(len(projfile.GetProjections()) > 0)
297    
298    
299    class TestProjFileWithInvalidParameters(unittest.TestCase,
300                                            support.FileLoadTestCase):
301    
302        file_extension = ".proj"
303        file_contents = '''\
304    <?xml version="1.0" encoding="UTF-8"?>
305    <!DOCTYPE projectionlist SYSTEM "projfile.dtd">
306    <projectionlist>
307        <projection name="Universal Transverse Mercator">
308            <parameter value="proj=utm"/>
309            <parameter value="ellps=clrk66"/>
310            <!-- an invalid zone number to trigger the parameter checking
311                 in the proj library -->
312            <parameter value="zone=1000"/>
313        </projection>
314        <projection name="Transverse Mercator">
315            <parameter value="proj=tmerc"/>
316            <parameter value="ellps=clrk66"/>
317            <parameter value="lat_0=90w"/>
318            <parameter value="lon_0=90w"/>
319            <parameter value="k=1"/>
320        </projection>
321    </projectionlist>
322    '''
323    
324        def setUp(self):
325            support.FileLoadTestCase.setUp(self)
326    
327        def test(self):
328            """Test reading a proj file with invalid parameters"""
329            projfile, warnings = resource.read_proj_file(self.filename())
330            projs = projfile.GetProjections()
331            self.assertEquals(len(projs), 1)
332            params = projs[0].GetAllParameters()[:]
333            params.sort()
334            self.assertEquals(params, ['ellps=clrk66', 'k=1', 'lat_0=90w',
335                                       'lon_0=90w', 'proj=tmerc'])
336            self.assertEquals(warnings,
337                           ['Error in projection "Universal Transverse Mercator":'
338                            ' invalid UTM zone number'])
339    
340    
341    
342  if __name__ == "__main__":  if __name__ == "__main__":
343      unittest.main()      unittest.main()
Legend:
Removed from v.752  
changed lines
  Added in v.1798
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26