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

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

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

revision 684 by jonathan, Tue Apr 15 21:55:12 2003 UTC revision 947 by jonathan, Tue May 20 15:27:19 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 60  contents_single_map = '''\ Line 60  contents_single_map = '''\
60  </session>  </session>
61  '''  '''
62    
63    contents_classified_map_v0_2 = '''\
64    <?xml version="1.0" encoding="UTF-8"?>
65    <!DOCTYPE session SYSTEM "thuban.dtd">
66    <session title="single map&amp;layer">
67            <map title="Test Map">
68                    <projection>
69                            <parameter value="zone=26"/>
70                            <parameter value="proj=utm"/>
71                            <parameter value="ellps=clrk66"/>
72                    </projection>
73                    <layer title="My Layer" stroke_width="1" fill="None"
74                        filename="../../Data/iceland/political.shp"
75                        stroke="#000000">
76                <classification field="POPYREG" field_type="string">
77                    <clnull>
78                        <cldata stroke="#000000" stroke_width="1" fill="None"/>
79                    </clnull>
80                    <clpoint value="1">
81                        <cldata stroke="#000000" stroke_width="2" fill="None"/>
82                    </clpoint>
83                    <clpoint value="1">
84                        <cldata stroke="#000000" stroke_width="10" fill="None"/>
85                    </clpoint>
86                </classification>
87            </layer>
88                    <layer title="My Layer 2" stroke_width="1" fill="None"
89                        filename="../../Data/iceland/political.shp"
90                        stroke="#000000">
91                <classification field="AREA" field_type="double">
92                    <clnull>
93                        <cldata stroke="#000000" stroke_width="2" fill="None"/>
94                    </clnull>
95                    <clrange min="0" max="1">
96                        <cldata stroke="#111111" stroke_width="1" fill="None"/>
97                    </clrange>
98                    <clpoint value=".5">
99                        <cldata stroke="#000000" stroke_width="1" fill="#111111"/>
100                    </clpoint>
101                    <clrange min="-1" max="0">
102                        <cldata stroke="#000000" stroke_width="1" fill="None"/>
103                    </clrange>
104                    <clpoint value="-.5">
105                        <cldata stroke="#000000" stroke_width="1" fill="None"/>
106                    </clpoint>
107                </classification>
108            </layer>
109            </map>
110    </session>
111    '''
112    
113    contents_test_labels = '''\
114    <?xml version="1.0" encoding="UTF-8"?>
115    <!DOCTYPE session SYSTEM "thuban.dtd">
116    <session title="single map&amp;layer">
117            <map title="Test Map">
118                    <projection>
119                            <parameter value="zone=26"/>
120                            <parameter value="proj=utm"/>
121                            <parameter value="ellps=clrk66"/>
122                    </projection>
123                    <layer title="My Layer" stroke_width="1" fill="None"
124                        filename="../../Data/iceland/political.shp"
125                        stroke="#000000">
126                <classification field="POPYREG" field_type="string">
127                    <clnull label="hallo">
128                        <cldata stroke="#000000" stroke_width="1" fill="None"/>
129                    </clnull>
130                    <clpoint label="welt" value="1">
131                        <cldata stroke="#000000" stroke_width="2" fill="None"/>
132                    </clpoint>
133                </classification>
134            </layer>
135        </map>
136    </session>
137    '''
138    
139    contents_test_layer_projection = '''\
140    <?xml version="1.0" encoding="UTF-8"?>
141    <!DOCTYPE session SYSTEM "thuban.dtd">
142    <session title="single map&amp;layer">
143            <map title="Test Map">
144                    <projection>
145                            <parameter value="zone=26"/>
146                            <parameter value="proj=utm"/>
147                            <parameter value="ellps=clrk66"/>
148                    </projection>
149                    <layer title="My Layer" stroke_width="1" fill="None"
150                        filename="../../Data/iceland/political.shp"
151                        stroke="#000000">
152                        <projection name="hello">
153                            <parameter value="zone=13"/>
154                            <parameter value="proj=tmerc"/>
155                            <parameter value="ellps=clrk66"/>
156                        </projection>
157                <classification field="POPYREG" field_type="string">
158                    <clnull label="hallo">
159                        <cldata stroke="#000000" stroke_width="1" fill="None"/>
160                    </clnull>
161                    <clpoint label="welt" value="1">
162                        <cldata stroke="#000000" stroke_width="2" fill="None"/>
163                    </clpoint>
164                </classification>
165            </layer>
166                    <layer title="My Layer" stroke_width="1" fill="None"
167                        filename="../../Data/iceland/political.shp"
168                        stroke="#000000">
169                        <projection>
170                            <parameter value="proj=lcc"/>
171                            <parameter value="ellps=clrk66"/>
172                        </projection>
173            </layer>
174        </map>
175    </session>
176    '''
177    
178    contents_test_visible = '''\
179    <?xml version="1.0" encoding="UTF-8"?>
180    <!DOCTYPE session SYSTEM "thuban.dtd">
181    <session title="single map&amp;layer">
182            <map title="Test Map">
183                    <projection>
184                            <parameter value="zone=26"/>
185                            <parameter value="proj=utm"/>
186                            <parameter value="ellps=clrk66"/>
187                    </projection>
188                    <layer title="My Layer" stroke_width="1" fill="None"
189                        filename="../../Data/iceland/political.shp"
190                        stroke="#000000" visible="false">
191            </layer>
192        </map>
193    </session>
194    '''
195    
196    contents_test_rasterlayer = '''\
197    <?xml version="1.0" encoding="UTF-8"?>
198    <!DOCTYPE session SYSTEM "thuban.dtd">
199    <session title="single map&amp;layer">
200            <map title="Test Map">
201                    <rasterlayer title="My RasterLayer"
202                         filename="../../Data/iceland/island.tif"
203                         visible="false">
204            </rasterlayer>
205        </map>
206    </session>
207    '''
208    
209  class LoadSessionTest(unittest.TestCase, support.FileTestMixin):  class LoadSessionTest(unittest.TestCase, support.FileTestMixin):
210    
# Line 68  class LoadSessionTest(unittest.TestCase, Line 213  class LoadSessionTest(unittest.TestCase,
213          file = open(self.temp_file_name("load_singlelayer.thuban"), "w")          file = open(self.temp_file_name("load_singlelayer.thuban"), "w")
214          file.write(contents_single_map)          file.write(contents_single_map)
215          file.close()          file.close()
216    
217            file = open(self.temp_file_name("load_classified_v0_2.thuban"), "w")
218            file.write(contents_classified_map_v0_2)
219            file.close()
220    
221            file = open(self.temp_file_name("load_labels.thuban"), "w")
222            file.write(contents_test_labels)
223            file.close()
224    
225            file = open(self.temp_file_name("load_layerproj.thuban"), "w")
226            file.write(contents_test_layer_projection)
227            file.close()
228    
229            file = open(self.temp_file_name("load_visible.thuban"), "w")
230            file.write(contents_test_visible)
231            file.close()
232    
233            file = open(self.temp_file_name("load_rasterlayer.thuban"), "w")
234            file.write(contents_test_rasterlayer)
235            file.close()
236    
237          self.session = None          self.session = None
238    
239      def tearDown(self):      def tearDown(self):
240          if self.session is not None:          if self.session is not None:
241              self.session.Destroy()              self.session.Destroy()
242            self.session = None
243    
244      def testSingleLayer(self):      def testSingleLayer(self):
245          """Load a session with a single map with a single layer"""          """Load a session with a single map with a single layer"""
# Line 105  class LoadSessionTest(unittest.TestCase, Line 272  class LoadSessionTest(unittest.TestCase,
272                                                       "political.shp")))                                                       "political.shp")))
273          eq(layer.GetClassification().GetDefaultFill(), Color.Transparent)          eq(layer.GetClassification().GetDefaultFill(), Color.Transparent)
274          eq(layer.GetClassification().GetDefaultLineColor().hex(), "#000000")          eq(layer.GetClassification().GetDefaultLineColor().hex(), "#000000")
275            eq(layer.Visible(), True)
276    
277          self.session.Destroy()          self.session.Destroy()
278          self.session = None          self.session = None
279    
280        def testLayerVisibility(self):
281            """Test that the visible flag is correctly loaded for a layer."""
282    
283            eq = self.assertEquals
284            session = load_session(self.temp_file_name("load_visible.thuban"))
285            self.session = session
286            maps = session.Maps()
287            eq(len(maps), 1)
288            map = maps[0]
289            layers = map.Layers()
290            eq(len(layers), 1)
291            layer = layers[0]
292    
293            eq(layer.Visible(), False)
294    
295      def testClassification(self):      def testClassification(self):
296          """Load a session with a map and classified layers."""          """Load a session with a map and classified layers."""
297    
298          eq = self.assertEquals          session = load_session(self.temp_file_name("load_classified_v0_2.thuban"))
         session = load_session("../Data/iceland_sample_test.thuban")  
299          self.session = session          self.session = session
300    
301          map = self.session.Maps()[0] # only one map in the sample          map = self.session.Maps()[0] # only one map in the sample
302    
303          layers = map.Layers()          expected = [("My Layer", 2,
         eq(len(layers), 3)  
   
         expected = (("political", 0,  
                         [("default", (), "",  
                             ("#000000", 1, "#c0c0c0"))]),  
                     ("roads-line", 5,  
304                          [("default", (), "",                          [("default", (), "",
305                              ("#000000", 1, "None")),                              ("#000000", 1, "None")),
306                           ("range", (0.001, 0.3996), "short roads",                           ("single", "1", "",
307                              ("#ffffff", 1, "#ffffff")),                              ("#000000", 2, "None")),
308                           ("range", (0.3996, 0.7982), "less short",                           ("single", "1", "",
309                              ("#ccf4cc", 1, "#ccf4cc")),                              ("#000000", 10, "None"))]),
310                           ("range", (0.7982, 1.1968), "bit longer",                       ("My Layer 2", 4,
311                              ("#99ea99", 1, "#99ea99")),                           [("default", (), "",
312                           ("range", (1.1968, 1.5954), "where's the end?",                              ("#000000", 2, "None")),
313                              ("#66e066", 1, "#66e066")),                            ("range", (0, 1), "",
314                           ("range", (1.5954, 1.994), "long roads",                              ("#111111", 1, "None")),
315                              ("#33d633", 1, "#33d633"))]),                            ("single", .5, "",
316                      ("something else", 6,                              ("#000000", 1, "#111111")),
317                          [("default", (), "",                            ("range", (-1, 0), "",
318                              ("#000000", 1, "None")),                              ("#000000", 1, "None")),
319                           ("single", ("BUILDING"),  "",                            ("single", -.5, "",
320                              ("#0000ff", 1, "#0000ff")),                              ("#000000", 1, "None"))])]
                          ("single", ("FARM"), "",  
                             ("#00aaff", 1, "#00aaff")),  
                          ("single", ("HUT"), "",  
                             ("#00ffaa", 1, "#00ffaa")),  
                          ("single", ("LIGHTHOUSE"), "",  
                             ("#00ff00", 1, "#00ff00")),  
                          ("single", ("OTHER/UNKNOWN"), "",  
                             ("#aaff00", 1, "#aaff00")),  
                          ("single", ("RUINS"), "",  
                             ("#ffaa00", 1, "#ffaa00"))]))  
321    
322          eq(len(layers), len(expected))          self.TestLayers(map.Layers(), expected)
323    
324        def TestLayers(self, layers, expected):
325    
326          TITLE = 0          TITLE = 0
327          NUM_GROUPS = 1          NUM_GROUPS = 1
# Line 163  class LoadSessionTest(unittest.TestCase, Line 331  class LoadSessionTest(unittest.TestCase,
331          GROUP_LABEL = 2          GROUP_LABEL = 2
332          GROUP_PROPS = 3          GROUP_PROPS = 3
333    
334            eq = self.assertEquals
335    
336            eq(len(layers), len(expected))
337    
338          for layer, data in zip(layers, expected):          for layer, data in zip(layers, expected):
339              eq(layer.Title(), data[TITLE])              eq(layer.Title(), data[TITLE])
340    
# Line 194  class LoadSessionTest(unittest.TestCase, Line 366  class LoadSessionTest(unittest.TestCase,
366    
367                  i += 1                  i += 1
368    
369        def testLabels(self):
370            """Load a session and test for reading the group labels."""
371    
372            eq = self.assertEquals
373            session = load_session(self.temp_file_name("load_labels.thuban"))
374            self.session = session
375    
376            map = self.session.Maps()[0] # only one map in the sample
377    
378            expected = [("My Layer", 1,
379                            [("default", (), "hallo",
380                                ("#000000", 1, "None")),
381                             ("single", "1", "welt",
382                                ("#000000", 2, "None"))])]
383    
384            self.TestLayers(map.Layers(), expected)
385    
386        def testLayerProjection(self):
387            eq = self.assertEquals
388            neq = self.assertNotEqual
389    
390            session = load_session(self.temp_file_name("load_layerproj.thuban"))
391            self.session = session
392    
393            map = self.session.Maps()[0] # only one map in the sample
394    
395            layers = map.Layers() # two layers in the sample
396    
397            # test layer with a named projection
398            proj = layers[0].GetProjection()
399            neq(proj, None)
400            eq(proj.GetName(), "hello")
401            eq(proj.GetParameter("proj"), "tmerc")
402            eq(proj.GetParameter("zone"), "13")
403            eq(proj.GetParameter("ellps"), "clrk66")
404    
405            # test layer with an unnamed projection
406            proj = layers[1].GetProjection()
407            neq(proj, None)
408            eq(proj.GetName(), "Unknown")
409            eq(proj.GetParameter("proj"), "lcc")
410            eq(proj.GetParameter("ellps"), "clrk66")
411    
412        def testRasterLayer(self):
413            eq = self.assertEquals
414            neq = self.assertNotEqual
415    
416            session = load_session(self.temp_file_name("load_rasterlayer.thuban"))
417            self.session = session
418    
419            map = self.session.Maps()[0] # only one map in the sample
420    
421            layer = map.Layers()[0] # one layer in the sample
422    
423            eq(layer.Title(), "My RasterLayer")
424            self.failIf(layer.Visible())
425            self.failUnless(filenames_equal(layer.GetImageFilename(),
426                                            os.path.join(self.temp_dir(),
427                                                         os.pardir, os.pardir,
428                                                         "Data", "iceland",
429                                                         "island.tif")))
430    
431  if __name__ == "__main__":  if __name__ == "__main__":
432      unittest.main()      unittest.main()
433    
Legend:
Removed from v.684  
changed lines
  Added in v.947
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26