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

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

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

revision 530 by jonathan, Wed Mar 12 19:58:00 2003 UTC revision 1200 by jonathan, Fri Jun 13 15:04:44 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 24  from xml.sax import make_parser, ErrorHa Line 24  from xml.sax import make_parser, ErrorHa
24  import support  import support
25  support.initthuban()  support.initthuban()
26    
27  from Thuban.Model.save import save_session  from Thuban.Model.save import XMLWriter, save_session
28  from Thuban.Model.session import Session  from Thuban.Model.session import Session
29  from Thuban.Model.map import Map  from Thuban.Model.map import Map
30  from Thuban.Model.layer import Layer  from Thuban.Model.layer import Layer, RasterLayer
31  from Thuban.Model.proj import Projection  from Thuban.Model.proj import Projection
32    
33    from Thuban.Model.classification import ClassGroupSingleton, ClassGroupRange, \
34        ClassGroupProperties
35    
36    from Thuban.Model.range import Range
37    
38    
39  class SaxEventLister(xml.sax.handler.ContentHandler):  class SaxEventLister(xml.sax.handler.ContentHandler):
40    
# Line 70  def sax_eventlist(data): Line 75  def sax_eventlist(data):
75    
76      return handler.eventlist      return handler.eventlist
77    
78    class XMLWriterTest(unittest.TestCase):
79    
80        def testEncode(self):
81            """Test XMLWriter.encode"""
82            writer = XMLWriter()
83            eq = self.assertEquals
84    
85            eq(writer.encode("hello world"), "hello world")
86            eq(writer.encode(unicode("hello world")), unicode("hello world"))
87    
88            eq(writer.encode("\x80\x90\xc2\x100"),
89                             "\xc2\x80\xc2\x90\xc3\x82\x100")
90            eq(writer.encode(u"\x80\x90\xc2\x100"),
91                             "\xc2\x80\xc2\x90\xc3\x82\x100")
92            eq(writer.encode(u"\xFF5E"), "\xc3\xbf5E")
93    
94            eq(writer.encode('&"\'<>'), "&amp;&quot;&apos;&lt;&gt;")
95            eq(writer.encode(unicode('&"\'<>')), "&amp;&quot;&apos;&lt;&gt;")
96    
97  class SaveSessionTest(unittest.TestCase, support.FileTestMixin):  class SaveSessionTest(unittest.TestCase, support.FileTestMixin):
98    
99      def compare_xml(self, xml1, xml2):      def compare_xml(self, xml1, xml2):
# Line 100  class SaveSessionTest(unittest.TestCase, Line 124  class SaveSessionTest(unittest.TestCase,
124          # use shapefile from the example data          # use shapefile from the example data
125          shpfile = os.path.join(os.path.dirname(__file__),          shpfile = os.path.join(os.path.dirname(__file__),
126                                 os.pardir, "Data", "iceland", "political.shp")                                 os.pardir, "Data", "iceland", "political.shp")
127          layer = Layer("My Layer", shpfile)          layer = Layer("My Layer", session.OpenShapefile(shpfile))
128            map.AddLayer(layer)
129    
130            filename = self.temp_file_name("save_singlemap.thuban")
131            save_session(session, filename)
132    
133            file = open(filename)
134            written_contents = file.read()
135            file.close()
136            expected_template = '''<?xml version="1.0" encoding="UTF-8"?>
137            <!DOCTYPE session SYSTEM "thuban.dtd">
138            <session title="single map&amp;layer">
139                <map title="Test Map">
140                    <projection name="Unknown">
141                        <parameter value="zone=26"/>
142                        <parameter value="proj=utm"/>
143                        <parameter value="ellps=clrk66"/>
144                    </projection>
145                    <layer title="My Layer" filename="%s"
146                    fill="None" stroke="#000000" stroke_width="1" visible="%s"/>
147                </map>
148            </session>'''
149            
150            expected_contents = expected_template % \
151                (os.path.join("..", "..", "Data", "iceland", "political.shp"),
152                 "true")
153    
154            #print written_contents
155            #print "********************************************"
156            #print expected_contents
157            self.compare_xml(written_contents, expected_contents)
158    
159            layer.SetVisible(False)
160            save_session(session, filename)
161    
162            file = open(filename)
163            written_contents = file.read()
164            file.close()
165            expected_contents = expected_template % \
166                (os.path.join("..", "..", "Data", "iceland", "political.shp"),
167                 "false")
168    
169            #print written_contents
170            #print "********************************************"
171            #print expected_contents
172            self.compare_xml(written_contents, expected_contents)
173    
174            session.Destroy()
175    
176        def testLayerProjection(self):
177            # deliberately put an apersand in the title :)
178            session = Session("single map&layer")
179            proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])
180            map = Map("Test Map", projection = proj)
181            session.AddMap(map)
182            # use shapefile from the example data
183            shpfile = os.path.join(os.path.dirname(__file__),
184                                   os.pardir, "Data", "iceland", "political.shp")
185            layer = Layer("My Layer", session.OpenShapefile(shpfile))
186            proj = Projection(["proj=lcc", "ellps=clrk66"], "Layer Projection")
187            layer.SetProjection(proj)
188          map.AddLayer(layer)          map.AddLayer(layer)
189    
190          filename = self.temp_file_name("save_singlemap.thuban")          filename = self.temp_file_name("save_singlemap.thuban")
# Line 114  class SaveSessionTest(unittest.TestCase, Line 198  class SaveSessionTest(unittest.TestCase,
198          <!DOCTYPE session SYSTEM "thuban.dtd">          <!DOCTYPE session SYSTEM "thuban.dtd">
199          <session title="single map&amp;layer">          <session title="single map&amp;layer">
200              <map title="Test Map">              <map title="Test Map">
201                  <projection>                  <projection name="Unknown">
202                      <parameter value="zone=26"/>                      <parameter value="zone=26"/>
203                      <parameter value="proj=utm"/>                      <parameter value="proj=utm"/>
204                      <parameter value="ellps=clrk66"/>                      <parameter value="ellps=clrk66"/>
205                  </projection>                  </projection>
206                  <layer title="My Layer" filename="%s"                  <layer title="My Layer" filename="%s"
207                  fill="None" stroke="#000000" stroke_width="1"/>                  fill="None" stroke="#000000" stroke_width="1" visible="true">
208                        <projection name="Layer Projection">
209                            <parameter value="proj=lcc"/>
210                            <parameter value="ellps=clrk66"/>
211                        </projection>
212                    </layer>
213              </map>              </map>
214          </session>''' % os.path.join("..", "..", "Data", "iceland",          </session>''' % os.path.join("..", "..", "Data", "iceland",
215                                       "political.shp")                                       "political.shp")
# Line 129  class SaveSessionTest(unittest.TestCase, Line 218  class SaveSessionTest(unittest.TestCase,
218          #print expected_contents          #print expected_contents
219          self.compare_xml(written_contents, expected_contents)          self.compare_xml(written_contents, expected_contents)
220    
221            
222        def testRasterLayer(self):
223            # deliberately put an apersand in the title :)
224            session = Session("single map&layer")
225            map = Map("Test Map")
226            session.AddMap(map)
227            # use shapefile from the example data
228            imgfile = os.path.join(os.path.dirname(__file__),
229                                   os.pardir, "Data", "iceland", "island.tif")
230            layer = RasterLayer("My RasterLayer", imgfile)
231            map.AddLayer(layer)
232                                                                                    
233            filename = self.temp_file_name("save_singlemap.thuban")
234            save_session(session, filename)
235            session.Destroy()
236                                                                                    
237            file = open(filename)
238            written_contents = file.read()
239            file.close()
240            expected_contents = '''<?xml version="1.0" encoding="UTF-8"?>
241            <!DOCTYPE session SYSTEM "thuban.dtd">
242            <session title="single map&amp;layer">
243                <map title="Test Map">
244                    <rasterlayer title="My RasterLayer" filename="%s"
245                                 visible="true">
246                    </rasterlayer>
247                </map>
248            </session>''' % os.path.join(os.path.dirname(__file__),
249                                         os.pardir, "Data", "iceland",
250                                         "island.tif")
251            #print written_contents
252            #print "********************************************"
253            #print expected_contents
254            self.compare_xml(written_contents, expected_contents)
255    
256        def testClassifiedLayer(self):
257            """Save a session with a single map with a single layer
258               with a classificaton.
259            """
260            # deliberately put an apersand in the title :)
261            session = Session("single map&layer")
262            proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])
263            map = Map("Test Map", projection = proj)
264            session.AddMap(map)
265            # use shapefile from the example data
266            shpfile = os.path.join(os.path.dirname(__file__),
267                                   os.pardir, "Data", "iceland", "political.shp")
268            layer = Layer("My Layer", session.OpenShapefile(shpfile))
269            map.AddLayer(layer)
270    
271            clazz = layer.GetClassification()
272    
273            clazz.SetField("AREA")
274    
275            clazz.AppendGroup(ClassGroupSingleton(42,
276                                               ClassGroupProperties(),
277                                               "single"))
278            clazz.AppendGroup(ClassGroupSingleton("text",
279                                               ClassGroupProperties(),
280                                               "single-text"))
281    
282            clazz.AppendGroup(ClassGroupRange(0, 42,
283                                               ClassGroupProperties(),
284                                               "range"))
285    
286            range = ClassGroupRange(Range("[0;42]"))
287            range.SetProperties(ClassGroupProperties())
288            range.SetLabel("new-range")
289            clazz.AppendGroup(range)
290    
291            filename = self.temp_file_name("save_singlemap.thuban")
292            save_session(session, filename)
293    
294            file = open(filename)
295            written_contents = file.read()
296            file.close()
297            expected_template = '''<?xml version="1.0" encoding="UTF-8"?>
298            <!DOCTYPE session SYSTEM "thuban.dtd">
299            <session title="single map&amp;layer">
300                <map title="Test Map">
301                    <projection name="Unknown">
302                        <parameter value="zone=26"/>
303                        <parameter value="proj=utm"/>
304                        <parameter value="ellps=clrk66"/>
305                    </projection>
306                    <layer title="My Layer" filename="%s"
307                    fill="None" stroke="#000000" stroke_width="1" visible="%s">
308                        <classification field="AREA" field_type="double">
309                            <clnull label="">
310                                <cldata fill="None" stroke="#000000" stroke_width="1"/>
311                            </clnull>
312                            <clpoint value="42" label="single">
313                                <cldata fill="None" stroke="#000000" stroke_width="1"/>
314                            </clpoint>
315                            <clpoint value="text" label="single-text">
316                                <cldata fill="None" stroke="#000000" stroke_width="1"/>
317                            </clpoint>
318                            <clrange range="[0;42[" label="range">
319                                <cldata fill="None" stroke="#000000" stroke_width="1"/>
320                            </clrange>
321                            <clrange range="[0;42]" label="new-range">
322                                <cldata fill="None" stroke="#000000" stroke_width="1"/>
323                            </clrange>
324                        </classification>
325                    </layer>
326                </map>
327            </session>'''
328            
329            expected_contents = expected_template % \
330                (os.path.join("..", "..", "Data", "iceland", "political.shp"),
331                 "true")
332    
333            #print written_contents
334            #print "********************************************"
335            #print expected_contents
336            self.compare_xml(written_contents, expected_contents)
337    
338            session.Destroy()
339    
340    
341  if __name__ == "__main__":  if __name__ == "__main__":
342      # Fake the __file__ global because it's needed by a test      # Fake the __file__ global because it's needed by a test
343      import sys      import sys
344      __file__ = sys.argv[0]      __file__ = sys.argv[0]
345      unittest.main()      support.run_tests()
Legend:
Removed from v.530  
changed lines
  Added in v.1200
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26