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

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

Parent Directory Parent Directory | Revision Log Revision Log

Revision 723 - (hide annotations)
Thu Apr 24 15:31:53 2003 UTC (21 years, 10 months ago) by bh
Original Path: trunk/thuban/test/test_save.py
File MIME type: text/x-python
File size: 4439 byte(s) 
First step towards table management. Introduce a simple data
abstraction so that we replace the data a layer uses more easily
in the next step.

* Thuban/Model/data.py: New file with a simple data abstraction
that bundles shapefile and dbffile into one object.

* Thuban/Model/session.py (Session.OpenShapefile): New method to
open shapefiles and return a shape store object

* Thuban/Model/layer.py (Layer.__init__): Pass the data as a store
object instead of a shapefile filename. This introduces a new
instance variable store holding the datastore. For intermediate
backwards compatibility keep the old instance variables.
(open_shapefile): Removed. No longer needed with the shape store.
(Layer.SetShapeStore, Layer.ShapeStore): New methods to set and
get the shape store used by a layer.
(Layer.Destroy): No need to explicitly destroy the shapefile or
table anymore.

* Thuban/UI/mainwindow.py (MainWindow.AddLayer)
(MainWindow.AddLayer): Use the session's OpenShapefile method to
open shapefiles

* Thuban/Model/load.py (ProcessSession.start_layer): Use the
session's OpenShapefile method to open shapefiles

* test/test_classification.py
(TestClassification.test_classification): Use the session's
OpenShapefile method to open shapefiles and build the filename in
a more platform independed way

* test/test_layer.py (TestLayer.setUp, TestLayer.tearDown):
Implement to have a session to use in the tests
(TestLayer.test_arc_layer, TestLayer.test_polygon_layer)
(TestLayer.test_point_layer, TestLayer.test_empty_layer): Use the
session's OpenShapefile method to open shapefiles
(TestLayerLegend.setUp): Instantiate a session so that we can use
it to open shapefiles.
(TestLayerLegend.tearDown): Make sure that all references to
layers and session are removed otherwise we may get a resource
leak

* test/test_map.py (TestMapAddLayer.test_add_layer)
(TestMapWithContents.setUp): Instantiate a session so that we can
use it to open shapefiles.
(TestMapWithContents.tearDown): Make sure that all references to
layers, maps and sessions are removed otherwise we may get a
resource leak
("__main__"): use support.run_tests() so that more info about
uncollected garbage is printed

* test/test_save.py (SaveSessionTest.testSingleLayer): Use the
session's OpenShapefile method to open shapefiles
("__main__"): use support.run_tests() so that more info about
uncollected garbage is printed

* test/test_selection.py (TestSelection.tearDown): Make sure that
all references to the session and the selection are removed
otherwise we may get a resource leak
(TestSelection.get_layer): Instantiate a session so that we can
use it to open shapefiles.
("__main__"): use support.run_tests() so that more info about
uncollected garbage is printed

* test/test_session.py (TestSessionBase.tearDown)
(TestSessionWithContent.tearDown): Make sure that all references
to the session and layers are removed otherwise we may get a
resource leak
(TestSessionWithContent.setUp): Use the session's OpenShapefile
method to open shapefiles
1 bh 723 # Copyright (c) 2002, 2003 by Intevation GmbH
2 bh 292 # Authors:
3     # Bernhard Herzog <[email protected]>
4     #
5     # This program is free software under the GPL (>=v2)
6     # Read the file COPYING coming with Thuban for details.
7    
8     """
9     Test saving a thuban session as XML
10     """
11    
12     __version__ = "$Revision$"
13     # $Source$
14     # $Id$
15    
16     import os
17     import unittest
18     from StringIO import StringIO
19    
20     import xml.sax
21     import xml.sax.handler
22 jonathan 530 from xml.sax import make_parser, ErrorHandler, SAXNotRecognizedException
23 bh 292
24     import support
25     support.initthuban()
26    
27     from Thuban.Model.save import save_session
28     from Thuban.Model.session import Session
29     from Thuban.Model.map import Map
30     from Thuban.Model.layer import Layer
31     from Thuban.Model.proj import Projection
32    
33    
34     class SaxEventLister(xml.sax.handler.ContentHandler):
35    
36     def __init__(self):
37     self.eventlist = []
38    
39     def startElementNS(self, name, qname, attrs):
40     items = attrs.items()
41     items.sort()
42     self.eventlist.append(("start", name, qname, items))
43    
44     def endElementNS(self, name, qname):
45     self.eventlist.append(("end", name, qname))
46    
47    
48     def sax_eventlist(data):
49     """Return a list of SAX event generated for the XML data.
50     """
51     handler = SaxEventLister()
52     parser = make_parser()
53     parser.setContentHandler(handler)
54     parser.setErrorHandler(ErrorHandler())
55     parser.setFeature(xml.sax.handler.feature_namespaces, 1)
56    
57 jonathan 530 #
58     # see comment at the end of Thuban/Model/load.py
59     #
60     try:
61     parser.setFeature(xml.sax.handler.feature_validation, 0)
62     parser.setFeature(xml.sax.handler.feature_external_ges, 0)
63     parser.setFeature(xml.sax.handler.feature_external_pes, 0)
64     except SAXNotRecognizedException:
65     pass
66    
67 bh 292 inpsrc = xml.sax.InputSource()
68     inpsrc.setByteStream(StringIO(data))
69     parser.parse(inpsrc)
70    
71     return handler.eventlist
72    
73     class SaveSessionTest(unittest.TestCase, support.FileTestMixin):
74    
75     def compare_xml(self, xml1, xml2):
76     self.assertEquals(sax_eventlist(xml1), sax_eventlist(xml2))
77    
78     def testEmptySession(self):
79     """Save an empty session"""
80     session = Session("empty session")
81     filename = self.temp_file_name("save_emptysession.thuban")
82     save_session(session, filename)
83     session.Destroy()
84    
85     file = open(filename)
86     written_contents = file.read()
87     file.close()
88     self.compare_xml(written_contents,
89     '<?xml version="1.0" encoding="UTF-8"?>\n'
90     '<!DOCTYPE session SYSTEM "thuban.dtd">\n'
91     '<session title="empty session">\n</session>\n')
92    
93     def testSingleLayer(self):
94     """Save a session with a single map with a single layer"""
95     # deliberately put an apersand in the title :)
96     session = Session("single map&layer")
97     proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])
98     map = Map("Test Map", projection = proj)
99     session.AddMap(map)
100     # use shapefile from the example data
101     shpfile = os.path.join(os.path.dirname(__file__),
102     os.pardir, "Data", "iceland", "political.shp")
103 bh 723 layer = Layer("My Layer", session.OpenShapefile(shpfile))
104 bh 292 map.AddLayer(layer)
105    
106     filename = self.temp_file_name("save_singlemap.thuban")
107     save_session(session, filename)
108     session.Destroy()
109    
110     file = open(filename)
111     written_contents = file.read()
112     file.close()
113     expected_contents = '''<?xml version="1.0" encoding="UTF-8"?>
114     <!DOCTYPE session SYSTEM "thuban.dtd">
115     <session title="single map&amp;layer">
116     <map title="Test Map">
117     <projection>
118     <parameter value="zone=26"/>
119     <parameter value="proj=utm"/>
120     <parameter value="ellps=clrk66"/>
121     </projection>
122     <layer title="My Layer" filename="%s"
123     fill="None" stroke="#000000" stroke_width="1"/>
124     </map>
125     </session>''' % os.path.join("..", "..", "Data", "iceland",
126     "political.shp")
127 jonathan 494 #print written_contents
128     #print "********************************************"
129     #print expected_contents
130 bh 292 self.compare_xml(written_contents, expected_contents)
131    
132    
133    
134     if __name__ == "__main__":
135     # Fake the __file__ global because it's needed by a test
136     import sys
137     __file__ = sys.argv[0]
138 bh 723 support.run_tests()

Properties

Name Value
svn:eol-style native
svn:keywords Author Date Id Revision
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26