/[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 1259 by bh, Fri Jun 20 12:31:48 2003 UTC revision 1268 by bh, Fri Jun 20 16:10:12 2003 UTC
# Line 26  import xmlsupport Line 26  import xmlsupport
26  import support  import support
27  support.initthuban()  support.initthuban()
28    
29  from Thuban.Model.save import XMLWriter, save_session  import dbflib
30    
31    from Thuban.Lib.fileutil import relative_filename
32    from Thuban.Model.save import XMLWriter, save_session, sort_data_stores
33  from Thuban.Model.session import Session  from Thuban.Model.session import Session
34  from Thuban.Model.map import Map  from Thuban.Model.map import Map
35  from Thuban.Model.layer import Layer, RasterLayer  from Thuban.Model.layer import Layer, RasterLayer
36  from Thuban.Model.proj import Projection  from Thuban.Model.proj import Projection
37    from Thuban.Model.table import DBFTable
38    from Thuban.Model.transientdb import TransientJoinedTable
39    from Thuban.Model.data import DerivedShapeStore
40    
41  from Thuban.Model.classification import ClassGroupSingleton, ClassGroupRange, \  from Thuban.Model.classification import ClassGroupSingleton, ClassGroupRange, \
42      ClassGroupProperties      ClassGroupProperties
# Line 60  class XMLWriterTest(unittest.TestCase): Line 66  class XMLWriterTest(unittest.TestCase):
66  class SaveSessionTest(unittest.TestCase, support.FileTestMixin,  class SaveSessionTest(unittest.TestCase, support.FileTestMixin,
67                        xmlsupport.ValidationTest):                        xmlsupport.ValidationTest):
68    
69        dtd = "http://thuban.intevation.org/dtds/thuban-0.8.dtd"
70        thubanids = [((dtd, n), (None, "id")) for n in
71                     ["fileshapesource", "filetable", "jointable",
72                      "derivedshapesource"]]
73        thubanidrefs = [((dtd, n), (None, m)) for n, m in
74                        [("layer", "shapestore"),
75                         ("jointable", "left"),
76                         ("jointable", "right"),
77                         ("derivedshapesource", "table"),
78                         ("derivedshapesource", "shapesource")]]
79        del n, m, dtd
80    
81      def compare_xml(self, xml1, xml2):      def compare_xml(self, xml1, xml2):
82          self.assertEquals(xmlsupport.sax_eventlist(xml1),          if 0:
83                            xmlsupport.sax_eventlist(xml2))              for a, b in zip(sax_eventlist(xml1, self.thubanids,
84                                            self.thubanidrefs),
85                              sax_eventlist(xml2, self.thubanids,
86                                            self.thubanidrefs)):
87                  print a == b and ' ' or '*****'
88                  print a
89                  print b
90            self.assertEquals(xmlsupport.sax_eventlist(xml1, ids = self.thubanids,
91                                                       idrefs = self.thubanidrefs),
92                              xmlsupport.sax_eventlist(xml2, ids = self.thubanids,
93                                                       idrefs = self.thubanidrefs))
94    
95      def testEmptySession(self):      def testEmptySession(self):
96          """Save an empty session"""          """Save an empty session"""
# Line 76  class SaveSessionTest(unittest.TestCase, Line 104  class SaveSessionTest(unittest.TestCase,
104          file.close()          file.close()
105          self.compare_xml(written_contents,          self.compare_xml(written_contents,
106                           '<?xml version="1.0" encoding="UTF-8"?>\n'                           '<?xml version="1.0" encoding="UTF-8"?>\n'
107                           '<!DOCTYPE session SYSTEM "thuban.dtd">\n'                           '<!DOCTYPE session SYSTEM "thuban-0.8.dtd">\n'
108                           '<session title="empty session">\n</session>\n')                           '<session title="empty session" '
109             'xmlns="http://thuban.intevation.org/dtds/thuban-0.8.dtd">'
110                             '\n</session>\n')
111    
112          self.validate_data(written_contents)          self.validate_data(written_contents)
113    
# Line 101  class SaveSessionTest(unittest.TestCase, Line 131  class SaveSessionTest(unittest.TestCase,
131          written_contents = file.read()          written_contents = file.read()
132          file.close()          file.close()
133          expected_template = '''<?xml version="1.0" encoding="UTF-8"?>          expected_template = '''<?xml version="1.0" encoding="UTF-8"?>
134          <!DOCTYPE session SYSTEM "thuban.dtd">          <!DOCTYPE session SYSTEM "thuban-0.8.dtd">
135          <session title="single map&amp;layer">          <session title="single map&amp;layer"
136               xmlns="http://thuban.intevation.org/dtds/thuban-0.8.dtd">
137                <fileshapesource id="D1" filename="%s" filetype="shapefile"/>
138              <map title="Test Map">              <map title="Test Map">
139                  <projection name="Unknown">                  <projection name="Unknown">
140                      <parameter value="zone=26"/>                      <parameter value="zone=26"/>
141                      <parameter value="proj=utm"/>                      <parameter value="proj=utm"/>
142                      <parameter value="ellps=clrk66"/>                      <parameter value="ellps=clrk66"/>
143                  </projection>                  </projection>
144                  <layer title="My Layer" filename="%s"                  <layer title="My Layer" shapestore="D1"
145                  fill="None" stroke="#000000" stroke_width="1" visible="%s"/>                  fill="None" stroke="#000000" stroke_width="1" visible="%s"/>
146              </map>              </map>
147          </session>'''          </session>'''
148            
149          expected_contents = expected_template % \          expected_contents = expected_template % \
150              (os.path.join("..", "..", "Data", "iceland", "political.shp"),              (os.path.join("..", "..", "Data", "iceland", "political.shp"),
151               "true")               "true")
152    
         #print written_contents  
         #print "********************************************"  
         #print expected_contents  
153          self.compare_xml(written_contents, expected_contents)          self.compare_xml(written_contents, expected_contents)
154    
155          self.validate_data(written_contents)          self.validate_data(written_contents)
# Line 134  class SaveSessionTest(unittest.TestCase, Line 163  class SaveSessionTest(unittest.TestCase,
163          expected_contents = expected_template % \          expected_contents = expected_template % \
164              (os.path.join("..", "..", "Data", "iceland", "political.shp"),              (os.path.join("..", "..", "Data", "iceland", "political.shp"),
165               "false")               "false")
   
         #print written_contents  
         #print "********************************************"  
         #print expected_contents  
166          self.compare_xml(written_contents, expected_contents)          self.compare_xml(written_contents, expected_contents)
167          self.validate_data(written_contents)          self.validate_data(written_contents)
168    
169          session.Destroy()          session.Destroy()
170    
171      def testLayerProjection(self):      def testLayerProjection(self):
172            """Test saving layers with projections"""
173          # deliberately put an apersand in the title :)          # deliberately put an apersand in the title :)
174          session = Session("single map&layer")          session = Session("single map&layer")
175          proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])          proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])
# Line 157  class SaveSessionTest(unittest.TestCase, Line 183  class SaveSessionTest(unittest.TestCase,
183          layer.SetProjection(proj)          layer.SetProjection(proj)
184          map.AddLayer(layer)          map.AddLayer(layer)
185    
186          filename = self.temp_file_name("save_singlemap.thuban")          filename = self.temp_file_name("save_layerproj.thuban")
187          save_session(session, filename)          save_session(session, filename)
188          session.Destroy()          session.Destroy()
189    
# Line 165  class SaveSessionTest(unittest.TestCase, Line 191  class SaveSessionTest(unittest.TestCase,
191          written_contents = file.read()          written_contents = file.read()
192          file.close()          file.close()
193          expected_contents = '''<?xml version="1.0" encoding="UTF-8"?>          expected_contents = '''<?xml version="1.0" encoding="UTF-8"?>
194          <!DOCTYPE session SYSTEM "thuban.dtd">          <!DOCTYPE session SYSTEM "thuban-0.8.dtd">
195          <session title="single map&amp;layer">          <session title="single map&amp;layer"
196               xmlns="http://thuban.intevation.org/dtds/thuban-0.8.dtd">
197                <fileshapesource id="D1" filename="%s" filetype="shapefile"/>
198              <map title="Test Map">              <map title="Test Map">
199                  <projection name="Unknown">                  <projection name="Unknown">
200                      <parameter value="zone=26"/>                      <parameter value="zone=26"/>
201                      <parameter value="proj=utm"/>                      <parameter value="proj=utm"/>
202                      <parameter value="ellps=clrk66"/>                      <parameter value="ellps=clrk66"/>
203                  </projection>                  </projection>
204                  <layer title="My Layer" filename="%s"                  <layer title="My Layer" shapestore="D1"
205                  fill="None" stroke="#000000" stroke_width="1" visible="true">                  fill="None" stroke="#000000" stroke_width="1" visible="true">
206                      <projection name="Layer Projection">                      <projection name="Layer Projection">
207                          <parameter value="proj=lcc"/>                          <parameter value="proj=lcc"/>
# Line 209  class SaveSessionTest(unittest.TestCase, Line 237  class SaveSessionTest(unittest.TestCase,
237          written_contents = file.read()          written_contents = file.read()
238          file.close()          file.close()
239          expected_contents = '''<?xml version="1.0" encoding="UTF-8"?>          expected_contents = '''<?xml version="1.0" encoding="UTF-8"?>
240          <!DOCTYPE session SYSTEM "thuban.dtd">          <!DOCTYPE session SYSTEM "thuban-0.8.dtd">
241          <session title="single map&amp;layer">          <session title="single map&amp;layer"
242               xmlns="http://thuban.intevation.org/dtds/thuban-0.8.dtd">
243              <map title="Test Map">              <map title="Test Map">
244                  <rasterlayer title="My RasterLayer" filename="%s"                  <rasterlayer title="My RasterLayer" filename="%s"
245                               visible="true">                               visible="true">
246                  </rasterlayer>                  </rasterlayer>
247              </map>              </map>
248          </session>''' % os.path.join(os.path.dirname(__file__),          </session>''' % os.path.join(os.path.dirname(__file__),
249                                       os.pardir, "Data", "iceland",                                       os.pardir, "Data", "iceland",
250                                       "island.tif")                                       "island.tif")
251          #print written_contents          #print written_contents
# Line 268  class SaveSessionTest(unittest.TestCase, Line 297  class SaveSessionTest(unittest.TestCase,
297          written_contents = file.read()          written_contents = file.read()
298          file.close()          file.close()
299          expected_template = '''<?xml version="1.0" encoding="UTF-8"?>          expected_template = '''<?xml version="1.0" encoding="UTF-8"?>
300          <!DOCTYPE session SYSTEM "thuban.dtd">          <!DOCTYPE session SYSTEM "thuban-0.8.dtd">
301          <session title="single map&amp;layer">          <session title="single map&amp;layer"
302               xmlns="http://thuban.intevation.org/dtds/thuban-0.8.dtd">
303                <fileshapesource id="D1" filename="%s" filetype="shapefile"/>
304              <map title="Test Map">              <map title="Test Map">
305                  <projection name="Unknown">                  <projection name="Unknown">
306                      <parameter value="zone=26"/>                      <parameter value="zone=26"/>
307                      <parameter value="proj=utm"/>                      <parameter value="proj=utm"/>
308                      <parameter value="ellps=clrk66"/>                      <parameter value="ellps=clrk66"/>
309                  </projection>                  </projection>
310                  <layer title="My Layer" filename="%s"                  <layer title="My Layer" shapestore="D1"
311                  fill="None" stroke="#000000" stroke_width="1" visible="%s">                  fill="None" stroke="#000000" stroke_width="1" visible="%s">
312                      <classification field="AREA" field_type="double">                      <classification field="AREA" field_type="double">
313                          <clnull label="">                          <clnull label="">
# Line 312  class SaveSessionTest(unittest.TestCase, Line 343  class SaveSessionTest(unittest.TestCase,
343    
344          session.Destroy()          session.Destroy()
345    
346        def test_dbf_table(self):
347            """Test saving a session with a dbf table link"""
348            session = Session("a DBF Table session")
349            # use shapefile from the example data
350            dbffile = os.path.join(os.path.dirname(__file__),
351                                   os.pardir, "Data", "iceland", "political.dbf")
352            table = session.AddTable(DBFTable(dbffile))
353    
354            filename = self.temp_file_name("save_singletable.thuban")
355            save_session(session, filename)
356    
357            file = open(filename)
358            written_contents = file.read()
359            file.close()
360            expected_template = '''<?xml version="1.0" encoding="UTF-8"?>
361            <!DOCTYPE session SYSTEM "thuban-0.8.dtd">
362            <session title="a DBF Table session"
363               xmlns="http://thuban.intevation.org/dtds/thuban-0.8.dtd">
364                <filetable id="D1" filename="%s" filetype="DBF" title="political"/>
365            </session>'''
366    
367            expected_contents = expected_template % dbffile
368            self.compare_xml(written_contents, expected_contents)
369    
370        def test_joined_table(self):
371            """Test saving a session with joined table"""
372            # Create a simple table to use in the join
373            dbffile = self.temp_file_name("save_joinedtable.dbf")
374            dbf = dbflib.create(dbffile)
375            dbf.add_field("RDTYPE", dbflib.FTInteger, 10, 0)
376            dbf.add_field("TEXT", dbflib.FTString, 10, 0)
377            dbf.write_record(0, {'RDTYPE': 8, "TEXT": "foo"})
378            dbf.write_record(1, {'RDTYPE': 2, "TEXT": "bar"})
379            dbf.write_record(2, {'RDTYPE': 3, "TEXT": "baz"})
380            dbf.close()
381    
382            # Create the session and a map
383            session = Session("A Joined Table session")
384            try:
385                map = Map("Test Map")
386                session.AddMap(map)
387    
388                # Add the dbf file to the session
389                dbftable = session.AddTable(DBFTable(dbffile))
390    
391                # Create a layer with the shapefile to use in the join
392                shpfile = os.path.join(os.path.abspath(os.path.dirname(__file__)),
393                                       os.pardir, "Data", "iceland",
394                                       "roads-line.shp")
395                layer = Layer("My Layer", session.OpenShapefile(shpfile))
396                map.AddLayer(layer)
397    
398                # Do the join
399                store = layer.ShapeStore()
400                #for col in store.Table().Columns():
401                #    print col.name
402                joined = TransientJoinedTable(session.TransientDB(),
403                                              store.Table(), "RDLNTYPE",
404                                              dbftable, "RDTYPE")
405                store = session.AddShapeStore(DerivedShapeStore(store, joined))
406                layer.SetShapeStore(store)
407    
408                # Save the session
409                filename = self.temp_file_name("save_joinedtable.thuban")
410                save_session(session, filename)
411    
412                # Read it back and compare
413                file = open(filename)
414                written_contents = file.read()
415                file.close()
416                expected_template = '''<?xml version="1.0" encoding="UTF-8"?>
417                <!DOCTYPE session SYSTEM "thuban-0.8.dtd">
418                <session title="A Joined Table session"
419                 xmlns="http://thuban.intevation.org/dtds/thuban-0.8.dtd">
420                    <fileshapesource filename="%(shpfile)s"
421                                     filetype="shapefile" id="D142197204"/>
422                    <filetable filename="%(dbffile)s"
423                               title="save_joinedtable"
424                               filetype="DBF" id="D141881756"/>
425                    <jointable id="D142180284"
426                               title="Join of roads-line and save_joinedtable"
427                               leftcolumn="RDLNTYPE" left="D142197204"
428                               rightcolumn="RDTYPE" right="D141881756"/>
429                    <derivedshapesource id="D141915644"
430                                        table="D142180284"
431                                        shapesource="D142197204"/>
432                    <map title="Test Map">
433                        <layer title="My Layer"
434                               shapestore="D141915644" visible="true"
435                               stroke="#000000" stroke_width="1" fill="None"/>
436                    </map>
437                </session>'''
438    
439                expected_contents = expected_template % {
440                    "dbffile": relative_filename(self.temp_dir(), dbffile),
441                    "shpfile": relative_filename(self.temp_dir(), shpfile)
442                    }
443                self.compare_xml(written_contents, expected_contents)
444            finally:
445                session.Destroy()
446                session = None
447    
448    
449    class MockDataStore:
450    
451        """A very simple data store that only has dependencies"""
452    
453        def __init__(self, name, *dependencies):
454            self.name = name
455            self.dependencies = dependencies
456    
457        def __repr__(self):
458            return self.name
459    
460        def Dependencies(self):
461            return self.dependencies
462    
463    
464    class TestStoreSort(unittest.TestCase):
465    
466        def check_sort(self, containers, sorted):
467            """Check whether the list of data containers is sorted"""
468            # check whether sorted is in the right order
469            seen = {}
470            for container in sorted:
471                self.failIf(id(container) in seen,
472                            "Container %r at least twice in %r" % (container,
473                                                                   sorted))
474                for dep in container.Dependencies():
475                    self.assert_(id(dep) in seen,
476                                 "Dependency %r of %r not yet seen" % (dep,
477                                                                       container))
478                seen[id(container)] = 1
479            # check whether all of containers is in sorted
480            for container in containers:
481                self.assert_(id(container) in seen,
482                             "Container %r in containers but not in sorted")
483            self.assertEquals(len(containers), len(sorted))
484    
485        def test_sort_data_stores(self):
486            """Test Thuban.Model.save.sort_data_stores"""
487            d1 = MockDataStore("d1")
488            d2 = MockDataStore("d2")
489            d3 = MockDataStore("d3", d1)
490            d4 = MockDataStore("d4", d1, d3)
491    
492            containers = [d4, d1, d2, d3]
493            self.check_sort(containers, sort_data_stores(containers))
494            containers = [d1, d3, d2, d4]
495            self.check_sort(containers, sort_data_stores(containers))
496    
497    
498    
499  if __name__ == "__main__":  if __name__ == "__main__":
500      # Fake the __file__ global because it's needed by a test      # Fake the __file__ global because it's needed by a test
Legend:
Removed from v.1259  
changed lines
  Added in v.1268
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26