WIP-pyshapelib-bramz/test/test_proj.py

# Copyright (c) 2002, 2003 by Intevation GmbH
# Authors:
# Bernhard Herzog <[email protected]>
#
# This program is free software under the GPL (>=v2)
# Read the file COPYING coming with Thuban for details.

"""
Test the Thuban-specific Projection class
"""

__version__ = "$Revision$"
# $Source$
# $Id$

import unittest
import os

import xmlsupport
import support
support.initthuban()

from Thuban import _
from Thuban.Model.proj import Projection, ProjFile, \
     PROJ_UNITS_METERS, PROJ_UNITS_DEGREES
from Thuban.Model.messages import PROJECTION_ADDED, PROJECTION_REMOVED, \
     PROJECTION_REPLACED
import Thuban.Model.resource as resource

from xmlsupport import sax_eventlist

from xml.sax import SAXParseException


class TestProjection(unittest.TestCase, support.FloatComparisonMixin):

    """Test cases for the Thuban-specific Projection class
    """

    def test(self):
        """Test Projection"""
        params = ["zone=26", "proj=utm", "ellps=clrk66"]
        proj = Projection(params)
        self.assertEquals(proj.params, params)

        # It's not clear whether this value is really the correct one
        # but a test failure here probably still means a bug somewhere
        self.assertFloatSeqEqual(proj.Forward(0, 0),
                                 [3623101.8103431347, 0.0],
                                 epsilon = 1e-5)
        self.assertFloatSeqEqual(proj.Inverse(3623101.8103431347, 0.0),
                                 [-0.00065775699878736467, 0])

        self.assertFloatSeqEqual(proj.ForwardBBox((0, 0, 2, 2)),
                                 (3620891.3077618643, 0.0,
                                  3875381.8535437919, 252962.10480170773),
                                 epsilon = 1e-5)

        # GetName()
        self.assertEquals(proj.GetName(), _("Unknown"))

        # GetParameter()
        self.assertEquals(proj.GetParameter("zone"), "26")
        self.assertEquals(proj.GetParameter("proj"), "utm")
        self.assertEquals(proj.GetParameter("ellps"), "clrk66")
        self.assertEquals(proj.GetParameter("hallo"), "")

        # GetAllParameters()
        self.assertEquals(proj.GetAllParameters(), params)

        # GetName()
        proj = Projection(params, "MyName")
        self.assertEquals(proj.GetName(), "MyName")

    def test_get_parameter_without_equals_sign(self):
        """Test Projection.GetParameter() for a parameter without '=' sign"""
        proj = Projection(["proj=utm", "zone=34", "south", "ellps=clrk66"])
        # The Projection class pretends that for parameters specified
        # without a value the value is the same as the parameter name.
        self.assertEquals(proj.GetParameter("south"), "south")

    def test_get_projection_units_geo(self):
        """Test Projection.GetProjectedUnits() for geographic projection.
        Test for the alias 'longlat' as well.
        """
        proj = Projection(["proj=latlong", "to_meter=0.017453292519943295",
                           "ellps=clrk66"])
        self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_DEGREES)
        proj = Projection(["proj=longlat", "to_meter=0.017453292519943295",
                           "ellps=clrk66"])
        self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_DEGREES)

    def test_get_projection_units_normal(self):
        """Test Projection.GetProjectedUnits() for normal projection"""
        proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])
        self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_METERS)

    def test_label(self):
        """Test Projection.Label() without epsg"""
        proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"],
                           name = "My Projection")
        self.assertEquals(proj.Label(), "My Projection")

    def test_label_epsg(self):
        """Test Projection.Label() with epsg"""
        proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"],
                           name = "My Projection", epsg="42")
        self.assertEquals(proj.Label(), "EPSG    42 My Projection")

    def test_epsgcode_for_non_epsg_projection(self):
        """Test Projection.EPSGCode() without epsg"""
        proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"],
                           name = "My Projection")
        self.assertEquals(proj.EPSGCode(), None)

    def test_epsgcode_for_real_epsg_projection(self):
        """Test Projection.EPSGCode() with epsg"""
        proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"],
                           name = "My Projection", epsg="42")
        self.assertEquals(proj.EPSGCode(), "42")


class TestProjFileSimple:

    def test_init(self):
        """Test ProjFile coinstructor"""
        proj_file = ProjFile("some_filename")
        self.assertEquals(proj_file.GetFilename(), "some_filename")
        self.assertEquals(len(proj_file.GetProjections()), 0)

    def test_set_filename(self):
        """Test ProjFile.SetFilename()"""
        proj_file = ProjFile("some_filename")
        proj.SetFilename("other_name")
        self.assertEquals(proj_file.GetFilename(), "other_name")


class TestProjFile(unittest.TestCase, support.SubscriberMixin):

    """Test cases for ProjFile objects"""

    def setUp(self):
        self.clear_messages()
        self.proj0 = Projection(["proj=tmerc", "ellps=clrk66"])
        self.proj1 = Projection(["proj=utm", "ellps=clrk66"])
        self.proj2 = Projection(["proj=lcc", "ellps=clrk66",
                                 "lat_1=0", "lat_2=20"])
        self.proj_file = ProjFile("some_filename")
        for msg in [PROJECTION_ADDED, PROJECTION_REMOVED, PROJECTION_REPLACED]:
            self.proj_file.Subscribe(msg, self.subscribe_with_params, msg)

    def tearDown(self):
        self.clear_messages()
        self.proj_file.Destroy()

    def test_add_remove(self):
        """Test ProjFile.Add() and ProjFile.Remove()"""
        self.proj_file.Add(self.proj0)
        self.proj_file.Add(self.proj1)
        self.assertEquals(self.proj_file.GetProjections(),
                          [self.proj0, self.proj1])
        self.check_messages([(self.proj0, PROJECTION_ADDED),
                             (self.proj1, PROJECTION_ADDED)])
        self.clear_messages()

        self.proj_file.Remove(self.proj0)
        self.assertEquals(self.proj_file.GetProjections(), [self.proj1])
        self.check_messages([(self.proj0, PROJECTION_REMOVED)])

    def test_remove_non_existing(self):
        """Test ProjFile.Remove(<proj not in projfile>)"""
        self.assertRaises(ValueError, self.proj_file.Remove, self.proj0)
        # Nothing happened, so no messages should have been sent
        self.check_messages([])

    def test_replace(self):
        """Test ProjFile.Replace()"""
        self.proj_file.Add(self.proj0)
        self.proj_file.Add(self.proj1)
        self.clear_messages()

        # Replace()
        self.proj_file.Replace(self.proj0, self.proj2)
        self.assertEquals(self.proj_file.GetProjections(),
                          [self.proj2, self.proj1])
        self.check_messages([(self.proj0, self.proj2, PROJECTION_REPLACED)])

    def test_replace_non_existing(self):
        """Test ProjFile.Replace(<proj not in projfile>, <some proj>)"""
        self.proj_file.Add(self.proj0)
        self.proj_file.Add(self.proj1)
        self.clear_messages()
        self.assertRaises(ValueError,
                          self.proj_file.Replace, self.proj2, self.proj0)
        # All projections should still be there
        self.assertEquals(self.proj_file.GetProjections(),
                          [self.proj0, self.proj1])
        # Nothing happened, so no messages should have been sent
        self.check_messages([])


class ProjFileTest(unittest.TestCase, support.FileTestMixin,
                   xmlsupport.ValidationTest):

    """Base class for the proj file tests that read or write files"""

    def filename(self):
        """Return the filename for the test"""
        return self.temp_file_name(self.id() + ".proj")


class ProjFileReadTests(ProjFileTest):

    """Test read ProjFile objects from files

    The files only cover error handling and the system projection file.
    """

    def test_read_non_existing_file(self):
        """Test read_proj_file with non-existing file"""
        self.assertRaises(IOError,
                          resource.read_proj_file,
                          self.temp_file_name("nonexistent.proj"))

    def test_read_unreadable_file(self):
        """Test read_proj_file with unreadable file

        As currently written this only works on unix-like systems and
        not e.g. on MS Windows.
        """
        filename = self.filename()
        file = open(filename, "w")
        file.close()
        os.chmod(filename, 0200) # write-only
        self.assertRaises(IOError, resource.read_proj_file, filename)

    def test_read_empty_file(self):
        """Test read_proj_file with empty file"""
        filename = self.filename()
        file = open(filename, "w")
        file.close()

        self.assertRaises(SAXParseException, resource.read_proj_file, filename)

    def test_get_system_proj_file(self):
        """Test resource.get_system_proj_file(DEFAULT_PROJ_FILE)

        This is primarily to test whether the system proj file contains
        invalid projection paramers and whether the proj file is not
        empty
        """
        projfile, warnings\
                  = resource.get_system_proj_file(resource.DEFAULT_PROJ_FILE)
        self.assertEquals(warnings, [])
        self.assert_(len(projfile.GetProjections()) > 0)

        # see whether it got cached and we get the same projfile object
        # when we read the file again
        projfile2, warnings \
                   = resource.get_system_proj_file(resource.DEFAULT_PROJ_FILE)
        self.assert_(projfile is projfile2)


class WriteProjFileTests(ProjFileTest):

    """Test cases for writing proj files"""

    def compare_xml(self, xml1, xml2):
        self.assertEquals(sax_eventlist(xml1), sax_eventlist(xml2))

    def doTestWrite(self, projfile, expected):
        filename = self.filename()

        resource.write_proj_file(projfile)

        file = open(filename)
        written_contents = file.read()
        file.close()
        self.compare_xml(written_contents, expected)
        self.validate_data(written_contents)
        self.validate_data(expected)

    def test_write(self):
        """Test write_proj_file"""
        pf = ProjFile(self.filename())
        pf.Add(Projection(['proj=tmerc', 'ellps=clrk66',
                           'lat_0=90w', 'lon_0=90w', 'k=1'],
                          "Transverse Mercator",))
        pf.Add(Projection(["proj=tmerc",
                           "lat_0=0.000000000", "lon_0=-62.000000000",
                           "k=0.999500", "x_0=400000.000", "y_0=0.000",
                           "ellps=clrk80", "units=m"],
                          "Anguilla 1957 / British West Indies Grid",
                          epsg="200"))
        file_contents = '''<?xml version="1.0" encoding="UTF-8"?>
        <!DOCTYPE projectionlist SYSTEM "projfile.dtd">
        <projectionlist>
            <projection name="Transverse Mercator">
                <parameter value="proj=tmerc"/>
                <parameter value="ellps=clrk66"/>
                <parameter value="lat_0=90w"/>
                <parameter value="lon_0=90w"/>
                <parameter value="k=1"/>
            </projection>
            <projection epsg="200"
                        name="Anguilla 1957 / British West Indies Grid">
                <parameter value="proj=tmerc"/>
                <parameter value="lat_0=0.000000000"/>
                <parameter value="lon_0=-62.000000000"/>
                <parameter value="k=0.999500"/>
                <parameter value="x_0=400000.000"/>
                <parameter value="y_0=0.000"/>
                <parameter value="ellps=clrk80"/>
                <parameter value="units=m"/>
            </projection>
        </projectionlist>
        '''
        self.doTestWrite(pf, file_contents)

    def test_write_empty_file(self):
        """Test write empty ProjFile"""
        pf = ProjFile(self.filename())
        file_contents = '''<?xml version="1.0" encoding="UTF-8"?>
        <!DOCTYPE projectionlist SYSTEM "projfile.dtd">
        <projectionlist>
        </projectionlist>
        '''
        self.doTestWrite(pf, file_contents)


class ProjFileLoadTestCase(support.FileLoadTestCase):

    """Base class for the test cases that read specific test files"""

    file_extension = ".proj"

    def tearDown(self):
        """Clear the cache explicitly"""
        resource.clear_proj_file_cache()


class TestLoadingProjFile(ProjFileLoadTestCase):

    file_contents = '''\
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE projectionlist SYSTEM "projfile.dtd">
<projectionlist>
    <projection name="Transverse Mercator">
        <parameter value="proj=tmerc"/>
        <parameter value="ellps=clrk66"/>
        <parameter value="lat_0=90w"/>
        <parameter value="lon_0=90w"/>
        <parameter value="k=1"/>
    </projection>
    <projection epsg="200" name="Anguilla 1957 / British West Indies Grid">
        <parameter value="proj=tmerc"/>
        <parameter value="lat_0=0.000000000"/>
        <parameter value="lon_0=-62.000000000"/>
        <parameter value="k=0.999500"/>
        <parameter value="x_0=400000.000"/>
        <parameter value="y_0=0.000"/>
        <parameter value="ellps=clrk80"/>
        <parameter value="units=m"/>
    </projection>
</projectionlist>
'''

    def check_projection(self, proj, label, parameters):
        """Check the values of the proj's label and parameters"""
        self.assertEquals(proj.Label(), label)
        params = proj.GetAllParameters()[:]
        params.sort()
        self.assertEquals(params, parameters)

    def test(self):
        projfile, warnings = resource.read_proj_file(self.filename())
        # no warnings
        self.assertEquals(warnings, [])

        # There are two projections
        projs = projfile.GetProjections()
        self.assertEquals(len(projs), 2)

        self.check_projection(projs[0],
                              "Transverse Mercator",
                              ['ellps=clrk66', 'k=1', 'lat_0=90w', 'lon_0=90w',
                               'proj=tmerc'])
        self.check_projection(projs[1],
                         "EPSG   200 Anguilla 1957 / British West Indies Grid",
                              ["ellps=clrk80", "k=0.999500",
                               "lat_0=0.000000000", "lon_0=-62.000000000",
                               "proj=tmerc", "units=m",
                               "x_0=400000.000", "y_0=0.000"])

    def test_caching(self):
        # test whether the projfile cache works
        projfile, warnings = resource.read_proj_file(self.filename())
        projfile2, warnings = resource.read_proj_file(self.filename())

        # Both projfiles should be the same object
        self.assert_(projfile2 is projfile)

        # If we clear the cache we should get a new one.
        resource.clear_proj_file_cache()
        projfile3, warnings = resource.read_proj_file(self.filename())
        self.assert_(projfile3 is not projfile)


class TestLoadingProjFileWithEmptyProjectionlist(ProjFileLoadTestCase):

    file_contents = '''\
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE projectionlist SYSTEM "projfile.dtd">
<projectionlist>
</projectionlist>
'''

    def test(self):
        projfile, warnings = resource.read_proj_file(self.filename())
        # no warnings
        self.assertEquals(warnings, [])

        # There are no projections
        self.assertEquals(len(projfile.GetProjections()), 0)


class TestProjFileWithInvalidParameters(ProjFileLoadTestCase):

    file_contents = '''\
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE projectionlist SYSTEM "projfile.dtd">
<projectionlist>
    <projection name="Universal Transverse Mercator">
        <parameter value="proj=utm"/>
        <parameter value="ellps=clrk66"/>
        <!-- an invalid zone number to trigger the parameter checking
             in the proj library -->
        <parameter value="zone=1000"/>
    </projection>
    <projection name="Transverse Mercator">
        <parameter value="proj=tmerc"/>
        <parameter value="ellps=clrk66"/>
        <parameter value="lat_0=90w"/>
        <parameter value="lon_0=90w"/>
        <parameter value="k=1"/>
    </projection>
</projectionlist>
'''

    def setUp(self):
        support.FileLoadTestCase.setUp(self)

    def test(self):
        """Test reading a proj file with invalid parameters"""
        projfile, warnings = resource.read_proj_file(self.filename())
        projs = projfile.GetProjections()
        self.assertEquals(len(projs), 1)
        params = projs[0].GetAllParameters()[:]
        params.sort()
        self.assertEquals(params, ['ellps=clrk66', 'k=1', 'lat_0=90w',
                                   'lon_0=90w', 'proj=tmerc'])
        self.assertEquals(warnings,
                       ['Error in projection "Universal Transverse Mercator":'
                        ' invalid UTM zone number'])


if __name__ == "__main__":
    support.run_tests()
1	bh	1259	# Copyright (c) 2002, 2003 by Intevation GmbH
2	bh	333	# 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 the Thuban-specific Projection class
10			"""
11
12			__version__ = "$Revision$"
13			# $Source$
14			# $Id$
15
16			import unittest
17	jonathan	719	import os
18	bh	333
19	bh	1756	import xmlsupport
20	bh	333	import support
21			support.initthuban()
22
23	bh	1684	from Thuban import _
24	bh	1793	from Thuban.Model.proj import Projection, ProjFile, \
25			PROJ_UNITS_METERS, PROJ_UNITS_DEGREES
26	bh	1825	from Thuban.Model.messages import PROJECTION_ADDED, PROJECTION_REMOVED, \
27			PROJECTION_REPLACED
28	jonathan	698	import Thuban.Model.resource as resource
29	bh	333
30	bh	1259	from xmlsupport import sax_eventlist
31	jonathan	698
32	jonathan	719	from xml.sax import SAXParseException
33	jonathan	698
34	jonathan	719
35	bh	333	class TestProjection(unittest.TestCase, support.FloatComparisonMixin):
36
37			"""Test cases for the Thuban-specific Projection class
38			"""
39
40			def test(self):
41			"""Test Projection"""
42	jonathan	761	params = ["zone=26", "proj=utm", "ellps=clrk66"]
43			proj = Projection(params)
44			self.assertEquals(proj.params, params)
45	bh	333
46			# It's not clear whether this value is really the correct one
47			# but a test failure here probably still means a bug somewhere
48			self.assertFloatSeqEqual(proj.Forward(0, 0),
49			[3623101.8103431347, 0.0],
50			epsilon = 1e-5)
51			self.assertFloatSeqEqual(proj.Inverse(3623101.8103431347, 0.0),
52			[-0.00065775699878736467, 0])
53
54			self.assertFloatSeqEqual(proj.ForwardBBox((0, 0, 2, 2)),
55			(3620891.3077618643, 0.0,
56			3875381.8535437919, 252962.10480170773),
57			epsilon = 1e-5)
58
59	jonathan	761	# GetName()
60			self.assertEquals(proj.GetName(), _("Unknown"))
61
62			# GetParameter()
63	jonathan	714	self.assertEquals(proj.GetParameter("zone"), "26")
64			self.assertEquals(proj.GetParameter("proj"), "utm")
65			self.assertEquals(proj.GetParameter("ellps"), "clrk66")
66			self.assertEquals(proj.GetParameter("hallo"), "")
67	bh	333
68	jonathan	761	# GetAllParameters()
69			self.assertEquals(proj.GetAllParameters(), params)
70	bh	333
71	jonathan	761	# GetName()
72			proj = Projection(params, "MyName")
73			self.assertEquals(proj.GetName(), "MyName")
74
75	bh	1798	def test_get_parameter_without_equals_sign(self):
76			"""Test Projection.GetParameter() for a parameter without '=' sign"""
77			proj = Projection(["proj=utm", "zone=34", "south", "ellps=clrk66"])
78			# The Projection class pretends that for parameters specified
79			# without a value the value is the same as the parameter name.
80			self.assertEquals(proj.GetParameter("south"), "south")
81
82	bh	1793	def test_get_projection_units_geo(self):
83	jan	1888	"""Test Projection.GetProjectedUnits() for geographic projection.
84			Test for the alias 'longlat' as well.
85			"""
86	bh	1793	proj = Projection(["proj=latlong", "to_meter=0.017453292519943295",
87			"ellps=clrk66"])
88			self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_DEGREES)
89	jan	1888	proj = Projection(["proj=longlat", "to_meter=0.017453292519943295",
90			"ellps=clrk66"])
91			self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_DEGREES)
92	jonathan	761
93	bh	1793	def test_get_projection_units_normal(self):
94			"""Test Projection.GetProjectedUnits() for normal projection"""
95			proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"])
96			self.assertEquals(proj.GetProjectedUnits(), PROJ_UNITS_METERS)
97
98	bh	1818	def test_label(self):
99			"""Test Projection.Label() without epsg"""
100			proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"],
101			name = "My Projection")
102			self.assertEquals(proj.Label(), "My Projection")
103	bh	1793
104	bh	1818	def test_label_epsg(self):
105			"""Test Projection.Label() with epsg"""
106			proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"],
107			name = "My Projection", epsg="42")
108			self.assertEquals(proj.Label(), "EPSG 42 My Projection")
109	jonathan	698
110	bh	1818	def test_epsgcode_for_non_epsg_projection(self):
111			"""Test Projection.EPSGCode() without epsg"""
112			proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"],
113			name = "My Projection")
114			self.assertEquals(proj.EPSGCode(), None)
115	jonathan	698
116	bh	1818	def test_epsgcode_for_real_epsg_projection(self):
117			"""Test Projection.EPSGCode() with epsg"""
118			proj = Projection(["zone=26", "proj=utm", "ellps=clrk66"],
119			name = "My Projection", epsg="42")
120			self.assertEquals(proj.EPSGCode(), "42")
121	jonathan	698
122
123
124	bh	1821	class TestProjFileSimple:
125	bh	1813
126	bh	1821	def test_init(self):
127			"""Test ProjFile coinstructor"""
128			proj_file = ProjFile("some_filename")
129			self.assertEquals(proj_file.GetFilename(), "some_filename")
130			self.assertEquals(len(proj_file.GetProjections()), 0)
131	bh	1813
132	bh	1821	def test_set_filename(self):
133			"""Test ProjFile.SetFilename()"""
134			proj_file = ProjFile("some_filename")
135			proj.SetFilename("other_name")
136			self.assertEquals(proj_file.GetFilename(), "other_name")
137	bh	1813
138
139	bh	1825	class TestProjFile(unittest.TestCase, support.SubscriberMixin):
140	bh	1821
141	bh	1831	"""Test cases for ProjFile objects"""
142	jonathan	698
143	bh	1821	def setUp(self):
144	bh	1825	self.clear_messages()
145	bh	1821	self.proj0 = Projection(["proj=tmerc", "ellps=clrk66"])
146			self.proj1 = Projection(["proj=utm", "ellps=clrk66"])
147			self.proj2 = Projection(["proj=lcc", "ellps=clrk66",
148			"lat_1=0", "lat_2=20"])
149	bh	1825	self.proj_file = ProjFile("some_filename")
150			for msg in [PROJECTION_ADDED, PROJECTION_REMOVED, PROJECTION_REPLACED]:
151			self.proj_file.Subscribe(msg, self.subscribe_with_params, msg)
152	jonathan	742
153	bh	1827	def tearDown(self):
154			self.clear_messages()
155			self.proj_file.Destroy()
156
157	bh	1821	def test_add_remove(self):
158			"""Test ProjFile.Add() and ProjFile.Remove()"""
159	bh	1825	self.proj_file.Add(self.proj0)
160			self.proj_file.Add(self.proj1)
161			self.assertEquals(self.proj_file.GetProjections(),
162			[self.proj0, self.proj1])
163			self.check_messages([(self.proj0, PROJECTION_ADDED),
164			(self.proj1, PROJECTION_ADDED)])
165			self.clear_messages()
166	jonathan	742
167	bh	1825	self.proj_file.Remove(self.proj0)
168			self.assertEquals(self.proj_file.GetProjections(), [self.proj1])
169			self.check_messages([(self.proj0, PROJECTION_REMOVED)])
170
171	bh	1821	def test_remove_non_existing(self):
172			"""Test ProjFile.Remove(<proj not in projfile>)"""
173	bh	1825	self.assertRaises(ValueError, self.proj_file.Remove, self.proj0)
174			# Nothing happened, so no messages should have been sent
175			self.check_messages([])
176	jonathan	742
177	bh	1821	def test_replace(self):
178			"""Test ProjFile.Replace()"""
179	bh	1825	self.proj_file.Add(self.proj0)
180			self.proj_file.Add(self.proj1)
181			self.clear_messages()
182	jonathan	761
183	bh	1821	# Replace()
184	bh	1825	self.proj_file.Replace(self.proj0, self.proj2)
185			self.assertEquals(self.proj_file.GetProjections(),
186			[self.proj2, self.proj1])
187			self.check_messages([(self.proj0, self.proj2, PROJECTION_REPLACED)])
188	jonathan	742
189	bh	1821	def test_replace_non_existing(self):
190			"""Test ProjFile.Replace(<proj not in projfile>, <some proj>)"""
191	bh	1825	self.proj_file.Add(self.proj0)
192			self.proj_file.Add(self.proj1)
193			self.clear_messages()
194	bh	1821	self.assertRaises(ValueError,
195	bh	1825	self.proj_file.Replace, self.proj2, self.proj0)
196			# All projections should still be there
197			self.assertEquals(self.proj_file.GetProjections(),
198			[self.proj0, self.proj1])
199			# Nothing happened, so no messages should have been sent
200			self.check_messages([])
201	jonathan	742
202
203	bh	1821	class ProjFileTest(unittest.TestCase, support.FileTestMixin,
204			xmlsupport.ValidationTest):
205	jonathan	742
206	bh	1821	"""Base class for the proj file tests that read or write files"""
207	jonathan	742
208	bh	1821	def filename(self):
209			"""Return the filename for the test"""
210			return self.temp_file_name(self.id() + ".proj")
211	jonathan	742
212
213	bh	1821	class ProjFileReadTests(ProjFileTest):
214	jonathan	752
215	bh	1831	"""Test read ProjFile objects from files
216	jonathan	761
217	bh	1831	The files only cover error handling and the system projection file.
218			"""
219
220	bh	1811	def test_read_non_existing_file(self):
221			"""Test read_proj_file with non-existing file"""
222	jonathan	719	self.assertRaises(IOError,
223	bh	1811	resource.read_proj_file,
224			self.temp_file_name("nonexistent.proj"))
225	jonathan	719
226	bh	1811	def test_read_unreadable_file(self):
227			"""Test read_proj_file with unreadable file
228
229			As currently written this only works on unix-like systems and
230			not e.g. on MS Windows.
231			"""
232	bh	1813	filename = self.filename()
233	jonathan	719	file = open(filename, "w")
234			file.close()
235			os.chmod(filename, 0200) # write-only
236	jonathan	1167	self.assertRaises(IOError, resource.read_proj_file, filename)
237	jonathan	719
238	bh	1811	def test_read_empty_file(self):
239			"""Test read_proj_file with empty file"""
240	bh	1813	filename = self.filename()
241	jonathan	719	file = open(filename, "w")
242			file.close()
243
244	jonathan	1167	self.assertRaises(SAXParseException, resource.read_proj_file, filename)
245	jonathan	719
246	bh	1787	def test_get_system_proj_file(self):
247	bh	1933	"""Test resource.get_system_proj_file(DEFAULT_PROJ_FILE)
248	jonathan	698
249	bh	1787	This is primarily to test whether the system proj file contains
250			invalid projection paramers and whether the proj file is not
251			empty
252			"""
253	bh	1933	projfile, warnings\
254			= resource.get_system_proj_file(resource.DEFAULT_PROJ_FILE)
255	bh	1787	self.assertEquals(warnings, [])
256			self.assert_(len(projfile.GetProjections()) > 0)
257
258	bh	1831	# see whether it got cached and we get the same projfile object
259			# when we read the file again
260	bh	1933	projfile2, warnings \
261			= resource.get_system_proj_file(resource.DEFAULT_PROJ_FILE)
262	bh	1831	self.assert_(projfile is projfile2)
263	bh	1787
264	bh	1831
265	bh	1821	class WriteProjFileTests(ProjFileTest):
266	bh	1813
267			"""Test cases for writing proj files"""
268
269			def compare_xml(self, xml1, xml2):
270			self.assertEquals(sax_eventlist(xml1), sax_eventlist(xml2))
271
272	bh	1818	def doTestWrite(self, projfile, expected):
273	bh	1813	filename = self.filename()
274
275	bh	1818	resource.write_proj_file(projfile)
276	bh	1813
277			file = open(filename)
278			written_contents = file.read()
279			file.close()
280			self.compare_xml(written_contents, expected)
281			self.validate_data(written_contents)
282			self.validate_data(expected)
283
284			def test_write(self):
285			"""Test write_proj_file"""
286	bh	1818	pf = ProjFile(self.filename())
287			pf.Add(Projection(['proj=tmerc', 'ellps=clrk66',
288			'lat_0=90w', 'lon_0=90w', 'k=1'],
289			"Transverse Mercator",))
290			pf.Add(Projection(["proj=tmerc",
291			"lat_0=0.000000000", "lon_0=-62.000000000",
292			"k=0.999500", "x_0=400000.000", "y_0=0.000",
293			"ellps=clrk80", "units=m"],
294			"Anguilla 1957 / British West Indies Grid",
295			epsg="200"))
296			file_contents = '''<?xml version="1.0" encoding="UTF-8"?>
297			<!DOCTYPE projectionlist SYSTEM "projfile.dtd">
298			<projectionlist>
299			<projection name="Transverse Mercator">
300			<parameter value="proj=tmerc"/>
301			<parameter value="ellps=clrk66"/>
302			<parameter value="lat_0=90w"/>
303			<parameter value="lon_0=90w"/>
304			<parameter value="k=1"/>
305			</projection>
306			<projection epsg="200"
307			name="Anguilla 1957 / British West Indies Grid">
308			<parameter value="proj=tmerc"/>
309			<parameter value="lat_0=0.000000000"/>
310			<parameter value="lon_0=-62.000000000"/>
311			<parameter value="k=0.999500"/>
312			<parameter value="x_0=400000.000"/>
313			<parameter value="y_0=0.000"/>
314			<parameter value="ellps=clrk80"/>
315			<parameter value="units=m"/>
316			</projection>
317			</projectionlist>
318			'''
319			self.doTestWrite(pf, file_contents)
320	bh	1813
321			def test_write_empty_file(self):
322			"""Test write empty ProjFile"""
323	bh	1818	pf = ProjFile(self.filename())
324			file_contents = '''<?xml version="1.0" encoding="UTF-8"?>
325			<!DOCTYPE projectionlist SYSTEM "projfile.dtd">
326			<projectionlist>
327			</projectionlist>
328			'''
329			self.doTestWrite(pf, file_contents)
330	bh	1813
331
332	bh	1831	class ProjFileLoadTestCase(support.FileLoadTestCase):
333	bh	1818
334	bh	1831	"""Base class for the test cases that read specific test files"""
335
336	bh	1818	file_extension = ".proj"
337	bh	1831
338			def tearDown(self):
339			"""Clear the cache explicitly"""
340			resource.clear_proj_file_cache()
341
342
343			class TestLoadingProjFile(ProjFileLoadTestCase):
344
345	bh	1818	file_contents = '''\
346			<?xml version="1.0" encoding="UTF-8"?>
347			<!DOCTYPE projectionlist SYSTEM "projfile.dtd">
348			<projectionlist>
349			<projection name="Transverse Mercator">
350			<parameter value="proj=tmerc"/>
351			<parameter value="ellps=clrk66"/>
352			<parameter value="lat_0=90w"/>
353			<parameter value="lon_0=90w"/>
354			<parameter value="k=1"/>
355			</projection>
356			<projection epsg="200" name="Anguilla 1957 / British West Indies Grid">
357			<parameter value="proj=tmerc"/>
358			<parameter value="lat_0=0.000000000"/>
359			<parameter value="lon_0=-62.000000000"/>
360			<parameter value="k=0.999500"/>
361			<parameter value="x_0=400000.000"/>
362			<parameter value="y_0=0.000"/>
363			<parameter value="ellps=clrk80"/>
364			<parameter value="units=m"/>
365			</projection>
366			</projectionlist>
367			'''
368
369			def check_projection(self, proj, label, parameters):
370			"""Check the values of the proj's label and parameters"""
371			self.assertEquals(proj.Label(), label)
372			params = proj.GetAllParameters()[:]
373			params.sort()
374			self.assertEquals(params, parameters)
375
376			def test(self):
377			projfile, warnings = resource.read_proj_file(self.filename())
378			# no warnings
379			self.assertEquals(warnings, [])
380
381			# There are two projections
382			projs = projfile.GetProjections()
383			self.assertEquals(len(projs), 2)
384
385			self.check_projection(projs[0],
386			"Transverse Mercator",
387			['ellps=clrk66', 'k=1', 'lat_0=90w', 'lon_0=90w',
388			'proj=tmerc'])
389			self.check_projection(projs[1],
390			"EPSG 200 Anguilla 1957 / British West Indies Grid",
391			["ellps=clrk80", "k=0.999500",
392			"lat_0=0.000000000", "lon_0=-62.000000000",
393			"proj=tmerc", "units=m",
394			"x_0=400000.000", "y_0=0.000"])
395
396	bh	1831	def test_caching(self):
397			# test whether the projfile cache works
398			projfile, warnings = resource.read_proj_file(self.filename())
399			projfile2, warnings = resource.read_proj_file(self.filename())
400	bh	1818
401	bh	1831	# Both projfiles should be the same object
402			self.assert_(projfile2 is projfile)
403	bh	1818
404	bh	1831	# If we clear the cache we should get a new one.
405			resource.clear_proj_file_cache()
406			projfile3, warnings = resource.read_proj_file(self.filename())
407			self.assert_(projfile3 is not projfile)
408
409
410			class TestLoadingProjFileWithEmptyProjectionlist(ProjFileLoadTestCase):
411
412	bh	1818	file_contents = '''\
413			<?xml version="1.0" encoding="UTF-8"?>
414			<!DOCTYPE projectionlist SYSTEM "projfile.dtd">
415			<projectionlist>
416			</projectionlist>
417			'''
418
419			def test(self):
420			projfile, warnings = resource.read_proj_file(self.filename())
421			# no warnings
422			self.assertEquals(warnings, [])
423
424			# There are no projections
425			self.assertEquals(len(projfile.GetProjections()), 0)
426
427
428	bh	1831	class TestProjFileWithInvalidParameters(ProjFileLoadTestCase):
429	bh	1787
430			file_contents = '''\
431			<?xml version="1.0" encoding="UTF-8"?>
432			<!DOCTYPE projectionlist SYSTEM "projfile.dtd">
433			<projectionlist>
434			<projection name="Universal Transverse Mercator">
435			<parameter value="proj=utm"/>
436			<parameter value="ellps=clrk66"/>
437			<!-- an invalid zone number to trigger the parameter checking
438			in the proj library -->
439			<parameter value="zone=1000"/>
440			</projection>
441			<projection name="Transverse Mercator">
442			<parameter value="proj=tmerc"/>
443			<parameter value="ellps=clrk66"/>
444			<parameter value="lat_0=90w"/>
445			<parameter value="lon_0=90w"/>
446			<parameter value="k=1"/>
447			</projection>
448			</projectionlist>
449			'''
450
451			def setUp(self):
452			support.FileLoadTestCase.setUp(self)
453
454			def test(self):
455			"""Test reading a proj file with invalid parameters"""
456			projfile, warnings = resource.read_proj_file(self.filename())
457			projs = projfile.GetProjections()
458			self.assertEquals(len(projs), 1)
459			params = projs[0].GetAllParameters()[:]
460			params.sort()
461			self.assertEquals(params, ['ellps=clrk66', 'k=1', 'lat_0=90w',
462			'lon_0=90w', 'proj=tmerc'])
463			self.assertEquals(warnings,
464			['Error in projection "Universal Transverse Mercator":'
465			' invalid UTM zone number'])
466
467
468	bh	333	if __name__ == "__main__":
469	bh	1827	support.run_tests()
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision