/[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 524 by jonathan, Wed Mar 12 18:26:55 2003 UTC revision 1259 by bh, Fri Jun 20 12:31:48 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 19  from StringIO import StringIO Line 19  from StringIO import StringIO
19    
20  import xml.sax  import xml.sax
21  import xml.sax.handler  import xml.sax.handler
22  from xml.sax import make_parser, ErrorHandler  from xml.sax import make_parser, ErrorHandler, SAXNotRecognizedException
23    
24    import xmlsupport
25    
26  import support  import support
27  support.initthuban()  support.initthuban()
28    
29  from Thuban.Model.save import save_session  from Thuban.Model.save import XMLWriter, save_session
30  from Thuban.Model.session import Session  from Thuban.Model.session import Session
31  from Thuban.Model.map import Map  from Thuban.Model.map import Map
32  from Thuban.Model.layer import Layer  from Thuban.Model.layer import Layer, RasterLayer
33  from Thuban.Model.proj import Projection  from Thuban.Model.proj import Projection
34    
35    from Thuban.Model.classification import ClassGroupSingleton, ClassGroupRange, \
36        ClassGroupProperties
37    
38    from Thuban.Model.range import Range
39    
40    
41    class XMLWriterTest(unittest.TestCase):
42    
43  class SaxEventLister(xml.sax.handler.ContentHandler):      def testEncode(self):
44            """Test XMLWriter.encode"""
45            writer = XMLWriter()
46            eq = self.assertEquals
47    
48      def __init__(self):          eq(writer.encode("hello world"), "hello world")
49          self.eventlist = []          eq(writer.encode(unicode("hello world")), unicode("hello world"))
50    
51      def startElementNS(self, name, qname, attrs):          eq(writer.encode("\x80\x90\xc2\x100"),
52          items = attrs.items()                           "\xc2\x80\xc2\x90\xc3\x82\x100")
53          items.sort()          eq(writer.encode(u"\x80\x90\xc2\x100"),
54          self.eventlist.append(("start", name, qname, items))                           "\xc2\x80\xc2\x90\xc3\x82\x100")
55            eq(writer.encode(u"\xFF5E"), "\xc3\xbf5E")
     def endElementNS(self, name, qname):  
         self.eventlist.append(("end", name, qname))  
   
   
 def sax_eventlist(data):  
     """Return a list of SAX event generated for the XML data.  
     """  
     handler = SaxEventLister()  
     parser = make_parser()  
     parser.setContentHandler(handler)  
     parser.setErrorHandler(ErrorHandler())  
     parser.setFeature(xml.sax.handler.feature_namespaces, 1)  
     parser.setFeature(xml.sax.handler.feature_validation, 0)  
     parser.setFeature(xml.sax.handler.feature_external_ges, 0)  
     parser.setFeature(xml.sax.handler.feature_external_pes, 0)  
   
     inpsrc = xml.sax.InputSource()  
     inpsrc.setByteStream(StringIO(data))  
     parser.parse(inpsrc)  
56    
57      return handler.eventlist          eq(writer.encode('&"\'<>'), "&amp;&quot;&apos;&lt;&gt;")
58            eq(writer.encode(unicode('&"\'<>')), "&amp;&quot;&apos;&lt;&gt;")
59    
60  class SaveSessionTest(unittest.TestCase, support.FileTestMixin):  class SaveSessionTest(unittest.TestCase, support.FileTestMixin,
61                          xmlsupport.ValidationTest):
62    
63      def compare_xml(self, xml1, xml2):      def compare_xml(self, xml1, xml2):
64          self.assertEquals(sax_eventlist(xml1), sax_eventlist(xml2))          self.assertEquals(xmlsupport.sax_eventlist(xml1),
65                              xmlsupport.sax_eventlist(xml2))
66    
67      def testEmptySession(self):      def testEmptySession(self):
68          """Save an empty session"""          """Save an empty session"""
# Line 83  class SaveSessionTest(unittest.TestCase, Line 79  class SaveSessionTest(unittest.TestCase,
79                           '<!DOCTYPE session SYSTEM "thuban.dtd">\n'                           '<!DOCTYPE session SYSTEM "thuban.dtd">\n'
80                           '<session title="empty session">\n</session>\n')                           '<session title="empty session">\n</session>\n')
81    
82            self.validate_data(written_contents)
83    
84      def testSingleLayer(self):      def testSingleLayer(self):
85          """Save a session with a single map with a single layer"""          """Save a session with a single map with a single layer"""
86          # deliberately put an apersand in the title :)          # deliberately put an apersand in the title :)
# Line 93  class SaveSessionTest(unittest.TestCase, Line 91  class SaveSessionTest(unittest.TestCase,
91          # use shapefile from the example data          # use shapefile from the example data
92          shpfile = os.path.join(os.path.dirname(__file__),          shpfile = os.path.join(os.path.dirname(__file__),
93                                 os.pardir, "Data", "iceland", "political.shp")                                 os.pardir, "Data", "iceland", "political.shp")
94          layer = Layer("My Layer", shpfile)          layer = Layer("My Layer", session.OpenShapefile(shpfile))
95            map.AddLayer(layer)
96    
97            filename = self.temp_file_name("save_singlemap.thuban")
98            save_session(session, filename)
99    
100            file = open(filename)
101            written_contents = file.read()
102            file.close()
103            expected_template = '''<?xml version="1.0" encoding="UTF-8"?>
104            <!DOCTYPE session SYSTEM "thuban.dtd">
105            <session title="single map&amp;layer">
106                <map title="Test Map">
107                    <projection name="Unknown">
108                        <parameter value="zone=26"/>
109                        <parameter value="proj=utm"/>
110                        <parameter value="ellps=clrk66"/>
111                    </projection>
112                    <layer title="My Layer" filename="%s"
113                    fill="None" stroke="#000000" stroke_width="1" visible="%s"/>
114                </map>
115            </session>'''
116            
117            expected_contents = expected_template % \
118                (os.path.join("..", "..", "Data", "iceland", "political.shp"),
119                 "true")
120    
121            #print written_contents
122            #print "********************************************"
123            #print expected_contents
124            self.compare_xml(written_contents, expected_contents)
125    
126            self.validate_data(written_contents)
127    
128            layer.SetVisible(False)
129            save_session(session, filename)
130    
131            file = open(filename)
132            written_contents = file.read()
133            file.close()
134            expected_contents = expected_template % \
135                (os.path.join("..", "..", "Data", "iceland", "political.shp"),
136                 "false")
137    
138            #print written_contents
139            #print "********************************************"
140            #print expected_contents
141            self.compare_xml(written_contents, expected_contents)
142            self.validate_data(written_contents)
143    
144            session.Destroy()
145    
146        def testLayerProjection(self):
147            # deliberately put an apersand in the title :)
148            session = Session("single map&layer")
149            proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])
150            map = Map("Test Map", projection = proj)
151            session.AddMap(map)
152            # use shapefile from the example data
153            shpfile = os.path.join(os.path.dirname(__file__),
154                                   os.pardir, "Data", "iceland", "political.shp")
155            layer = Layer("My Layer", session.OpenShapefile(shpfile))
156            proj = Projection(["proj=lcc", "ellps=clrk66"], "Layer Projection")
157            layer.SetProjection(proj)
158          map.AddLayer(layer)          map.AddLayer(layer)
159    
160          filename = self.temp_file_name("save_singlemap.thuban")          filename = self.temp_file_name("save_singlemap.thuban")
# Line 107  class SaveSessionTest(unittest.TestCase, Line 168  class SaveSessionTest(unittest.TestCase,
168          <!DOCTYPE session SYSTEM "thuban.dtd">          <!DOCTYPE session SYSTEM "thuban.dtd">
169          <session title="single map&amp;layer">          <session title="single map&amp;layer">
170              <map title="Test Map">              <map title="Test Map">
171                  <projection>                  <projection name="Unknown">
172                      <parameter value="zone=26"/>                      <parameter value="zone=26"/>
173                      <parameter value="proj=utm"/>                      <parameter value="proj=utm"/>
174                      <parameter value="ellps=clrk66"/>                      <parameter value="ellps=clrk66"/>
175                  </projection>                  </projection>
176                  <layer title="My Layer" filename="%s"                  <layer title="My Layer" filename="%s"
177                  fill="None" stroke="#000000" stroke_width="1"/>                  fill="None" stroke="#000000" stroke_width="1" visible="true">
178                        <projection name="Layer Projection">
179                            <parameter value="proj=lcc"/>
180                            <parameter value="ellps=clrk66"/>
181                        </projection>
182                    </layer>
183              </map>              </map>
184          </session>''' % os.path.join("..", "..", "Data", "iceland",          </session>''' % os.path.join("..", "..", "Data", "iceland",
185                                       "political.shp")                                       "political.shp")
# Line 122  class SaveSessionTest(unittest.TestCase, Line 188  class SaveSessionTest(unittest.TestCase,
188          #print expected_contents          #print expected_contents
189          self.compare_xml(written_contents, expected_contents)          self.compare_xml(written_contents, expected_contents)
190    
191            self.validate_data(written_contents)
192    
193        def testRasterLayer(self):
194            # deliberately put an apersand in the title :)
195            session = Session("single map&layer")
196            map = Map("Test Map")
197            session.AddMap(map)
198            # use shapefile from the example data
199            imgfile = os.path.join(os.path.dirname(__file__),
200                                   os.pardir, "Data", "iceland", "island.tif")
201            layer = RasterLayer("My RasterLayer", imgfile)
202            map.AddLayer(layer)
203    
204            filename = self.temp_file_name("save_singlemap.thuban")
205            save_session(session, filename)
206            session.Destroy()
207    
208            file = open(filename)
209            written_contents = file.read()
210            file.close()
211            expected_contents = '''<?xml version="1.0" encoding="UTF-8"?>
212            <!DOCTYPE session SYSTEM "thuban.dtd">
213            <session title="single map&amp;layer">
214                <map title="Test Map">
215                    <rasterlayer title="My RasterLayer" filename="%s"
216                                 visible="true">
217                    </rasterlayer>
218                </map>
219            </session>''' % os.path.join(os.path.dirname(__file__),
220                                         os.pardir, "Data", "iceland",
221                                         "island.tif")
222            #print written_contents
223            #print "********************************************"
224            #print expected_contents
225            self.compare_xml(written_contents, expected_contents)
226    
227            self.validate_data(written_contents)
228    
229        def testClassifiedLayer(self):
230            """Save a session with a single map with a single layer
231               with a classificaton.
232            """
233            # deliberately put an apersand in the title :)
234            session = Session("single map&layer")
235            proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])
236            map = Map("Test Map", projection = proj)
237            session.AddMap(map)
238            # use shapefile from the example data
239            shpfile = os.path.join(os.path.dirname(__file__),
240                                   os.pardir, "Data", "iceland", "political.shp")
241            layer = Layer("My Layer", session.OpenShapefile(shpfile))
242            map.AddLayer(layer)
243    
244            clazz = layer.GetClassification()
245    
246            clazz.SetField("AREA")
247    
248            clazz.AppendGroup(ClassGroupSingleton(42,
249                                               ClassGroupProperties(),
250                                               "single"))
251            clazz.AppendGroup(ClassGroupSingleton("text",
252                                               ClassGroupProperties(),
253                                               "single-text"))
254    
255            clazz.AppendGroup(ClassGroupRange(0, 42,
256                                               ClassGroupProperties(),
257                                               "range"))
258    
259            range = ClassGroupRange(Range("[0;42]"))
260            range.SetProperties(ClassGroupProperties())
261            range.SetLabel("new-range")
262            clazz.AppendGroup(range)
263    
264            filename = self.temp_file_name("save_singlemap.thuban")
265            save_session(session, filename)
266    
267            file = open(filename)
268            written_contents = file.read()
269            file.close()
270            expected_template = '''<?xml version="1.0" encoding="UTF-8"?>
271            <!DOCTYPE session SYSTEM "thuban.dtd">
272            <session title="single map&amp;layer">
273                <map title="Test Map">
274                    <projection name="Unknown">
275                        <parameter value="zone=26"/>
276                        <parameter value="proj=utm"/>
277                        <parameter value="ellps=clrk66"/>
278                    </projection>
279                    <layer title="My Layer" filename="%s"
280                    fill="None" stroke="#000000" stroke_width="1" visible="%s">
281                        <classification field="AREA" field_type="double">
282                            <clnull label="">
283                                <cldata fill="None" stroke="#000000" stroke_width="1"/>
284                            </clnull>
285                            <clpoint value="42" label="single">
286                                <cldata fill="None" stroke="#000000" stroke_width="1"/>
287                            </clpoint>
288                            <clpoint value="text" label="single-text">
289                                <cldata fill="None" stroke="#000000" stroke_width="1"/>
290                            </clpoint>
291                            <clrange range="[0;42[" label="range">
292                                <cldata fill="None" stroke="#000000" stroke_width="1"/>
293                            </clrange>
294                            <clrange range="[0;42]" label="new-range">
295                                <cldata fill="None" stroke="#000000" stroke_width="1"/>
296                            </clrange>
297                        </classification>
298                    </layer>
299                </map>
300            </session>'''
301    
302            expected_contents = expected_template % \
303                (os.path.join("..", "..", "Data", "iceland", "political.shp"),
304                 "true")
305    
306            #print written_contents
307            #print "********************************************"
308            #print expected_contents
309            self.compare_xml(written_contents, expected_contents)
310    
311            self.validate_data(written_contents)
312    
313            session.Destroy()
314    
315    
316  if __name__ == "__main__":  if __name__ == "__main__":
317      # Fake the __file__ global because it's needed by a test      # Fake the __file__ global because it's needed by a test
318      import sys      import sys
319      __file__ = sys.argv[0]      __file__ = sys.argv[0]
320      unittest.main()      support.run_tests()
Legend:
Removed from v.524  
changed lines
  Added in v.1259
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26