wkbheader/tests.py at master · MazeMap/wkbheader · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
import unittest
import wkbheader
import codecs

'''
Testwkbs extracted from Postgis 2.0.3
'''

class WKBHeaderTest(unittest.TestCase):
    def setUp(self):
        self.point_wgs84 = codecs.decode(
                "0101000020E610000000000000000008400000000000001040",
                'hex_codec'
        )#point(3, 4), 4326 Little endian
        self.point_google = codecs.decode(
                "010100002031BF0D0000000000000008400000000000001040",
                'hex_codec'
        )#point(3, 4), 900913 Little endian
        self.point_nosrid = codecs.decode(
                "010100000000000000000008400000000000001040",
                'hex_codec'
        )#point(3, 4), Little endian no srid

        self.geocol_line_poly_wgs84 = codecs.decode(
                "01030000A0E61000000100000004000000000000000000"+
                "2440000000000000084000000000000000400000000000"+
                "0010400000000000001C40000000000000204000000000"+
                "0000084000000000000022400000000000002840000000"+
                "000000244000000000000008400000000000000040",
                'hex_codec'
        )
        self.geocol_line_poly_nosrid = codecs.decode(
                "0103000080010000000400000000000000000024400000"+
                "0000000008400000000000000040000000000000104000"+
                "00000000001C4000000000000020400000000000000840"+
                "0000000000002240000000000000284000000000000024"+
                "4000000000000008400000000000000040",
                'hex_codec'
        )

        self.point_nosrid_bigendian = codecs.decode(
                "000000000140000000000000004010000000000000",
                'hex_codec'
        )

    def tearDown(self):
        pass

    def test_get_endian(self):
        assert wkbheader.has_little_endian(self.point_nosrid)
        assert wkbheader.has_little_endian(self.point_google)
        assert not wkbheader.has_little_endian(self.point_nosrid_bigendian)

    def test_has_srid(self):
        assert wkbheader._has_srid(wkbheader._get_type(self.point_google, wkbheader._LITTLE_ENDIAN))
        assert not wkbheader._has_srid(wkbheader._get_type(self.point_nosrid, wkbheader._LITTLE_ENDIAN))

    def test_get_srid(self):
        assert wkbheader.get_srid(self.point_nosrid) is None
        assert wkbheader.get_srid(self.point_google) == 900913
        assert wkbheader.get_srid(self.point_wgs84) == 4326

    def test_endian_types(self):
        assert wkbheader.get_type_int(self.point_nosrid) == wkbheader.get_type_int(self.point_nosrid_bigendian)

    def test_malformed_wkb(self):
        malformed = codecs.encode(self.point_nosrid, 'hex_codec')
        with self.assertRaises(TypeError) as hexerr:
            wkbheader.has_little_endian(malformed)
            assert 'hex' in hexerr.exception.message
        with self.assertRaises(TypeError) as nohexerr:
            wkbheader.has_little_endian('foobar')
            assert 'hex' not in nohexerr.exception.message

    def test_drop_srid_from_type(self):
        point_with_srid_type = 536870913
        point_without_srid = 1
        assert wkbheader._drop_type_srid(point_with_srid_type) == point_without_srid

    def test_drop_srid(self):
        assert wkbheader.drop_srid(self.point_nosrid) == self.point_nosrid
        assert wkbheader.drop_srid(self.point_google) == self.point_nosrid
        assert wkbheader.drop_srid(self.point_wgs84) == self.point_nosrid

    def test_create_srid(self):
        assert wkbheader.set_srid(self.point_nosrid, 4326) == self.point_wgs84
        assert wkbheader.set_srid(self.point_nosrid, 900913) == self.point_google

    def test_replace_srid(self):
        assert wkbheader.set_srid(self.point_google, 4326) == self.point_wgs84
        assert wkbheader.set_srid(self.point_wgs84, 900913) == self.point_google

    def test_geocollection_change_srid(self):
        assert wkbheader.drop_srid(self.geocol_line_poly_wgs84) == self.geocol_line_poly_nosrid
        assert wkbheader.set_srid(self.geocol_line_poly_nosrid, 4326) == self.geocol_line_poly_wgs84


if __name__ == '__main__':
    unittest.main()