/[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 2552 by jonathan, Fri Jan 28 15:54:00 2005 UTC revision 2688 by frank, Fri Jun 30 12:27:20 2006 UTC
# Line 1  Line 1 
1  # Copyright (c) 2002, 2003, 2004 by Intevation GmbH  # Copyright (c) 2002, 2003, 2004, 2005 by Intevation GmbH
2  # Authors:  # Authors:
3  # Bernhard Herzog <[email protected]>  # Bernhard Herzog <[email protected]>
4  #  #
# Line 25  support.initthuban() Line 25  support.initthuban()
25    
26  import dbflib  import dbflib
27    
28    from Thuban import internal_from_unicode
29  from Thuban.Lib.fileutil import relative_filename  from Thuban.Lib.fileutil import relative_filename
30  from Thuban.Model.save import XMLWriter, save_session, sort_data_stores  from Thuban.Model.save import XMLWriter, save_session, sort_data_stores
31  from Thuban.Model.session import Session  from Thuban.Model.session import Session
# Line 33  from Thuban.Model.layer import Layer, Ra Line 34  from Thuban.Model.layer import Layer, Ra
34  from Thuban.Model.proj import Projection  from Thuban.Model.proj import Projection
35  from Thuban.Model.table import DBFTable  from Thuban.Model.table import DBFTable
36  from Thuban.Model.transientdb import TransientJoinedTable  from Thuban.Model.transientdb import TransientJoinedTable
37  from Thuban.Model.data import DerivedShapeStore  from Thuban.Model.data import DerivedShapeStore, SHAPETYPE_ARC
38    
39  from Thuban.Model.classification import ClassGroupSingleton, ClassGroupRange, \  from Thuban.Model.classification import ClassGroupSingleton, ClassGroupRange, \
40      ClassGroupProperties      ClassGroupPattern, ClassGroupProperties
41    
42  from Thuban.Model.range import Range  from Thuban.Model.range import Range
43    
# Line 53  class XMLWriterTest(unittest.TestCase): Line 54  class XMLWriterTest(unittest.TestCase):
54          eq(writer.encode("hello world"), "hello world")          eq(writer.encode("hello world"), "hello world")
55          eq(writer.encode(unicode("hello world")), unicode("hello world"))          eq(writer.encode(unicode("hello world")), unicode("hello world"))
56    
57          eq(writer.encode("\x80\x90\xc2\x100"),          eq(writer.encode(internal_from_unicode(u"\x80\x90\xc2\x100")),
58                           "\xc2\x80\xc2\x90\xc3\x82\x100")                           "\xc2\x80\xc2\x90\xc3\x82\x100")
59          eq(writer.encode(u"\x80\x90\xc2\x100"),          eq(writer.encode(u"\x80\x90\xc2\x100"),
60                           "\xc2\x80\xc2\x90\xc3\x82\x100")                           "\xc2\x80\xc2\x90\xc3\x82\x100")
61          eq(writer.encode(u"\xFF5E"), "\xc3\xbf5E")          eq(writer.encode(u"\xFF5E"), "\xc3\xbf5E")
62    
# Line 155  class SaveSessionTest(unittest.TestCase, Line 156  class SaveSessionTest(unittest.TestCase,
156                      <parameter value="datum=WGS84"/>                      <parameter value="datum=WGS84"/>
157                      <parameter value="units=m"/>                      <parameter value="units=m"/>
158                  </projection>                  </projection>
159                  <layer title="My Layer" shapestore="D1"                  <layer title="My Layer" shapestore="D1" visible="%s">
160                  fill="None" stroke="#000000" stroke_width="1" visible="%s"/>                      <classification>
161                            <clnull label="">
162                                <cldata fill="None" stroke="#000000"
163                                    stroke_width="1"/>
164                            </clnull>
165                        </classification>
166                    </layer>
167              </map>              </map>
168          </session>'''          </session>'''
169    
# Line 215  class SaveSessionTest(unittest.TestCase, Line 222  class SaveSessionTest(unittest.TestCase,
222                      <parameter value="proj=utm"/>                      <parameter value="proj=utm"/>
223                      <parameter value="ellps=clrk66"/>                      <parameter value="ellps=clrk66"/>
224                  </projection>                  </projection>
225                  <layer title="My Layer" shapestore="D1"                  <layer title="My Layer" shapestore="D1" visible="true">
                 fill="None" stroke="#000000" stroke_width="1" visible="true">  
226                      <projection name="Layer Projection">                      <projection name="Layer Projection">
227                          <parameter value="proj=lcc"/>                          <parameter value="proj=lcc"/>
228                          <parameter value="ellps=clrk66"/>                          <parameter value="ellps=clrk66"/>
229                          <parameter value="lat_1=0"/>                          <parameter value="lat_1=0"/>
230                          <parameter value="lat_2=20"/>                          <parameter value="lat_2=20"/>
231                      </projection>                      </projection>
232                        <classification>
233                            <clnull label="">
234                                <cldata fill="None" stroke="#000000"
235                                    stroke_width="1"/>
236                            </clnull>
237                        </classification>
238                  </layer>                  </layer>
239              </map>              </map>
240          </session>'''          </session>'''
# Line 234  class SaveSessionTest(unittest.TestCase, Line 246  class SaveSessionTest(unittest.TestCase,
246          self.validate_data(written_contents)          self.validate_data(written_contents)
247    
248      def testRasterLayer(self):      def testRasterLayer(self):
         # deliberately put an apersand in the title :)  
         session = Session("single map&layer")  
         map = Map("Test Map")  
         session.AddMap(map)  
         # use shapefile from the example data  
         imgfile = os.path.join(os.path.dirname(__file__),  
                                os.pardir, "Data", "iceland", "island.tif")  
         layer = RasterLayer("My RasterLayer", imgfile)  
         map.AddLayer(layer)  
249    
250          filename = self.temp_file_name("%s.thuban" % self.id())          MASK_NONE = RasterLayer.MASK_NONE
251          save_session(session, filename)          MASK_BIT = RasterLayer.MASK_BIT
252          session.Destroy()          MASK_ALPHA = RasterLayer.MASK_ALPHA
253    
254            for opacity, masktype, opname, maskname in \
255                [(1,  MASK_BIT,   '', ''),
256                 (.2, MASK_BIT,   'opacity="0.2"', ''),
257                 (1,  MASK_ALPHA, '',              'masktype="alpha"'),
258                 (.5, MASK_ALPHA, 'opacity="0.5"', 'masktype="alpha"'),
259                 (1,  MASK_NONE,  '',              'masktype="none"'),
260                 (0,  MASK_NONE,  'opacity="0"',   'masktype="none"') ]:
261    
         file = open(filename)  
         written_contents = file.read()  
         file.close()  
         expected_contents = '''<?xml version="1.0" encoding="UTF-8"?>  
         <!DOCTYPE session SYSTEM "thuban-1.1.dtd">  
         <session title="single map&amp;layer"  
            xmlns="http://thuban.intevation.org/dtds/thuban-1.1-dev.dtd">  
             <map title="Test Map">  
                 <rasterlayer title="My RasterLayer"  
                         filename="../../Data/iceland/island.tif"  
                         visible="true">  
                 </rasterlayer>  
             </map>  
         </session>'''  
         #print written_contents  
         #print "********************************************"  
         #print expected_contents  
         self.compare_xml(written_contents, expected_contents)  
262    
263          self.validate_data(written_contents)              # deliberately put an apersand in the title :)
264                session = Session("single map&layer")
265                map = Map("Test Map")
266                session.AddMap(map)
267                # use shapefile from the example data
268                imgfile = os.path.join(os.path.dirname(__file__),
269                                       os.pardir, "Data", "iceland", "island.tif")
270                layer = RasterLayer("My RasterLayer", imgfile)
271    
272                layer.SetOpacity(opacity)
273                layer.SetMaskType(masktype)
274    
275                map.AddLayer(layer)
276    
277                filename = self.temp_file_name("%s.thuban" % self.id())
278                save_session(session, filename)
279                session.Destroy()
280    
281                file = open(filename)
282                written_contents = file.read()
283                file.close()
284                expected_contents = '''<?xml version="1.0" encoding="UTF-8"?>
285                <!DOCTYPE session SYSTEM "thuban-1.1.dtd">
286                <session title="single map&amp;layer"
287                   xmlns="http://thuban.intevation.org/dtds/thuban-1.1-dev.dtd">
288                    <map title="Test Map">
289                        <rasterlayer title="My RasterLayer"
290                                filename="../../Data/iceland/island.tif"
291                                visible="true" %s %s>
292                        </rasterlayer>
293                    </map>
294                </session>''' % (opname, maskname)
295                #print written_contents
296                #print "********************************************"
297                #print expected_contents
298                self.compare_xml(written_contents, expected_contents)
299    
300                self.validate_data(written_contents)
301    
302      def testClassifiedLayer(self):      def testClassifiedLayer(self):
303          """Save a session with a single map with classifications"""          """Save a session with a single map with classifications"""
# Line 307  class SaveSessionTest(unittest.TestCase, Line 337  class SaveSessionTest(unittest.TestCase,
337          layer2.SetClassificationColumn("POPYCOUN")          layer2.SetClassificationColumn("POPYCOUN")
338    
339          # Classification with Latin 1 text          # Classification with Latin 1 text
340          clazz.AppendGroup(ClassGroupSingleton('\xe4\xf6\xfc', # ae, oe, ue          clazz.AppendGroup(ClassGroupSingleton(
341                                                ClassGroupProperties(),              internal_from_unicode(u'\xe4\xf6\xfc'), # ae, oe, ue
342                                                '\xdcml\xe4uts')) # Uemlaeuts              ClassGroupProperties(),
343                internal_from_unicode(u'\xdcml\xe4uts'))) # Uemlaeuts
344    
345            # Pattern
346            clazz.AppendGroup(ClassGroupPattern("BUI", ClassGroupProperties(),
347                                                "pattern"))
348    
349          filename = self.temp_file_name("%s.thuban" % self.id())          filename = self.temp_file_name("%s.thuban" % self.id())
350          save_session(session, filename)          save_session(session, filename)
# Line 331  class SaveSessionTest(unittest.TestCase, Line 365  class SaveSessionTest(unittest.TestCase,
365                      <parameter value="proj=utm"/>                      <parameter value="proj=utm"/>
366                      <parameter value="ellps=clrk66"/>                      <parameter value="ellps=clrk66"/>
367                  </projection>                  </projection>
368                  <layer title="My Layer" shapestore="D1"                  <layer title="My Layer" shapestore="D1" visible="true">
                 fill="None" stroke="#000000" stroke_width="1" visible="true">  
369                      <classification field="AREA" field_type="double">                      <classification field="AREA" field_type="double">
370                          <clnull label="">                          <clnull label="">
371                              <cldata fill="None" stroke="#000000" stroke_width="1"/>                              <cldata fill="None" stroke="#000000" stroke_width="1"/>
# Line 351  class SaveSessionTest(unittest.TestCase, Line 384  class SaveSessionTest(unittest.TestCase,
384                          </clrange>                          </clrange>
385                      </classification>                      </classification>
386                  </layer>                  </layer>
387                  <layer title="My Layer" shapestore="D1"                  <layer title="My Layer" shapestore="D1" visible="true">
                 fill="None" stroke="#000000" stroke_width="1" visible="true">  
388                      <classification field="POPYCOUN" field_type="string">                      <classification field="POPYCOUN" field_type="string">
389                          <clnull label="">                          <clnull label="">
390                              <cldata fill="None" stroke="#000000" stroke_width="1"/>                              <cldata fill="None" stroke="#000000" stroke_width="1"/>
# Line 361  class SaveSessionTest(unittest.TestCase, Line 393  class SaveSessionTest(unittest.TestCase,
393                               label="\xc3\x9cml\xc3\xa4uts">                               label="\xc3\x9cml\xc3\xa4uts">
394                              <cldata fill="None" stroke="#000000" stroke_width="1"/>                              <cldata fill="None" stroke="#000000" stroke_width="1"/>
395                          </clpoint>                          </clpoint>
396                            <clpattern pattern="BUI" label="pattern">
397                                <cldata fill="None" stroke="#000000" stroke_width="1"/>
398                            </clpattern>
399                      </classification>                      </classification>
400                  </layer>                  </layer>
401              </map>              </map>
# Line 466  class SaveSessionTest(unittest.TestCase, Line 501  class SaveSessionTest(unittest.TestCase,
501                                      shapesource="D142197204"/>                                      shapesource="D142197204"/>
502                  <map title="Test Map">                  <map title="Test Map">
503                      <layer title="My Layer"                      <layer title="My Layer"
504                             shapestore="D141915644" visible="true"                             shapestore="D141915644" visible="true">
505                             stroke="#000000" stroke_width="1" fill="None"/>                          <classification>
506                                <clnull label="">
507                                    <cldata fill="None" stroke="#000000"
508                                            stroke_width="1"/>
509                                </clnull>
510                            </classification>
511                        </layer>
512                  </map>                  </map>
513              </session>'''              </session>'''
514    
# Line 491  class SaveSessionTest(unittest.TestCase, Line 532  class SaveSessionTest(unittest.TestCase,
532              def _fetch_table_information(self):              def _fetch_table_information(self):
533                  # pretend that we've found a geometry column                  # pretend that we've found a geometry column
534                  self.geometry_column = "the_geom"                  self.geometry_column = "the_geom"
535                    # pretend this is a ARC shape type.
536                    self.shape_type = SHAPETYPE_ARC
537              def IDColumn(self):              def IDColumn(self):
538                  """Return an object with a name attribute with value 'gid'"""                  """Return an object with a name attribute with value 'gid'"""
539                  class dummycol:                  class dummycol:
# Line 529  class SaveSessionTest(unittest.TestCase, Line 572  class SaveSessionTest(unittest.TestCase,
572                                 id_column="gid" geometry_column="the_geom"/>                                 id_column="gid" geometry_column="the_geom"/>
573                  <map title="Test Map">                  <map title="Test Map">
574                      <layer title="Roads to Nowhere"                      <layer title="Roads to Nowhere"
575                             shapestore="roads" visible="true"                             shapestore="roads" visible="true">
576                             stroke="#000000" stroke_width="1" fill="None"/>                          <classification>
577                                <clnull label="">
578                                    <cldata fill="None" stroke="#000000"
579                                        stroke_width="1"/>
580                                </clnull>
581                            </classification>
582                        </layer>
583                  </map>                  </map>
584              </session>'''              </session>'''
585              self.compare_xml(written, expected)              self.compare_xml(written, expected)
Legend:
Removed from v.2552  
changed lines
  Added in v.2688
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26