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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
"""Module for testing the pp.geo_image module.
"""
import datetime
import unittest
import numpy as np
import mpop.imageo.geo_image as geo_image
class TestGeoImage(unittest.TestCase):
"""Class for testing pp.geo_image.
"""
def setUp(self):
"""Setup the test.
"""
time_slot = datetime.datetime(2009, 10, 8, 14, 30)
self.img = geo_image.GeoImage(np.zeros((512, 512), dtype = np.uint8),
area_id = "euro", time_slot = time_slot)
# def test_add_overlay(self):
# """Add overlay to the image.
# """
# self.img.add_overlay((1,1,1))
# model = np.array([[1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
# [1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
# [1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
# [0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
# [0.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
# [0.0, 0.0, 1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0],
# [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
# [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
# [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
# [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]])
# self.assert_(np.all(self.img.channels[0][300:310, 300:310] == model))
# def test_save_geotiff(self):
# """Save to geotiff format.
# """
# self.img.save("test.tif", compression = 0)
# self.assertEqual(str(os.popen("tiffdump test.tif").read()),
# "test.tif:\n"
# "Magic: 0x4949 <little-endian> Version: 0x2a\n"
# "Directory 0: offset 524688 (0x80190) next 0 (0)\n"
# "ImageWidth (256) SHORT (3) 1<512>\n"
# "ImageLength (257) SHORT (3) 1<512>\n"
# "BitsPerSample (258) SHORT (3) 2<8 8>\n"
# "Compression (259) SHORT (3) 1<1>\n"
# "Photometric (262) SHORT (3) 1<1>\n"
# "SamplesPerPixel (277) SHORT (3) 1<2>\n"
# "PlanarConfig (284) SHORT (3) 1<1>\n"
# "DateTime (306) ASCII (2) 20<2009:10:08 14:30:00\\0>\n"
# "TileWidth (322) SHORT (3) 1<256>\n"
# "TileLength (323) SHORT (3) 1<256>\n"
# "TileOffsets (324) LONG (4) 4<400 131472 262544 393616"
# ">\n"
# "TileByteCounts (325) LONG (4) 4<131072 131072 131072 "
# "131072>\n"
# "ExtraSamples (338) SHORT (3) 1<0>\n"
# "SampleFormat (339) SHORT (3) 2<1 1>\n"
# "33550 (0x830e) DOUBLE (12) 3<8000 8000 0>\n"
# "33922 (0x8482) DOUBLE (12) 6<0 0 0 -2.71718e+06 -1.47"
# "505e+06 0>\n"
# "34735 (0x87af) SHORT (3) 64<1 1 0 15 1024 0 1 1 1025 "
# "0 1 1 1026 34737 8 0 2048 0 1 4326 2049 34737 7 8 ..."
# ">\n"
# "34736 (0x87b0) DOUBLE (12) 5<60 14 1 0 0>\n"
# "34737 (0x87b1) ASCII (2) 16<unnamed|WGS 84|\\0>\n"
# "42112 (0xa480) ASCII (2) 154<<GDALMetadata>\\n <Item"
# " n ...>\n")
# os.remove("test.tif")
# self.img.save("test.tif", compression = 6)
# self.assertEqual(os.popen("tiffdump test.tif").read(),
# "test.tif:\n"
# "Magic: 0x4949 <little-endian> Version: 0x2a\n"
# "Directory 0: offset 1012 (0x3f4) next 0 (0)\n"
# "ImageWidth (256) SHORT (3) 1<512>\n"
# "ImageLength (257) SHORT (3) 1<512>\n"
# "BitsPerSample (258) SHORT (3) 2<8 8>\n"
# "Compression (259) SHORT (3) 1<8>\n"
# "Photometric (262) SHORT (3) 1<1>\n"
# "SamplesPerPixel (277) SHORT (3) 1<2>\n"
# "PlanarConfig (284) SHORT (3) 1<1>\n"
# "DateTime (306) ASCII (2) 20<2009:10:08 14:30:00\\0>\n"
# "Predictor (317) SHORT (3) 1<1>\n"
# "TileWidth (322) SHORT (3) 1<256>\n"
# "TileLength (323) SHORT (3) 1<256>\n"
# "TileOffsets (324) LONG (4) 4<412 562 712 862>\n"
# "TileByteCounts (325) LONG (4) 4<150 150 150 150>\n"
# "ExtraSamples (338) SHORT (3) 1<0>\n"
# "SampleFormat (339) SHORT (3) 2<1 1>\n"
# "33550 (0x830e) DOUBLE (12) 3<8000 8000 0>\n"
# "33922 (0x8482) DOUBLE (12) 6<0 0 0 -2.71718e+06 -1.47"
# "505e+06 0>\n"
# "34735 (0x87af) SHORT (3) 64<1 1 0 15 1024 0 1 1 1025 "
# "0 1 1 1026 34737 8 0 2048 0 1 4326 2049 34737 7 8 ..."
# ">\n"
# "34736 (0x87b0) DOUBLE (12) 5<60 14 1 0 0>\n"
# "34737 (0x87b1) ASCII (2) 16<unnamed|WGS 84|\\0>\n"
# "42112 (0xa480) ASCII (2) 154<<GDALMetadata>\\n <Item"
# " n ...>\n")
# os.remove("test.tif")
if __name__ == "__main__":
unittest.main()
|
