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
|
'''
Runs unit tests for iterators.
To run the unit tests, type the following from the system command line:
# python -m spectral.tests.iterators
'''
from __future__ import absolute_import, division, print_function, unicode_literals
import numpy as np
from numpy.testing import assert_allclose
import spectral as spy
from spectral.algorithms.algorithms import iterator, iterator_ij
from spectral.tests.spytest import SpyTest
class IteratorTest(SpyTest):
'''Tests various math functions.'''
def setup(self):
self.image = spy.open_image('92AV3C.lan')
self.gt = spy.open_image('92AV3GT.GIS').read_band(0)
def test_iterator_all(self):
'''Iteration over all pixels.'''
data = self.image.load()
pixels = data.reshape((-1, data.shape[-1]))
sum = np.sum(pixels, 0)
itsum = np.sum(np.array([x for x in iterator(data)]), 0)
assert_allclose(sum, itsum)
def test_iterator_nonzero(self):
'''Iteration over all non-background pixels.'''
data = self.image.load()
classes = self.gt.ravel()
pixels = data.reshape((-1, data.shape[-1]))
sum = np.sum(pixels[classes > 0], 0)
itsum = np.sum(np.array([x for x in iterator(data, self.gt)]), 0)
assert_allclose(sum, itsum)
def test_iterator_index(self):
'''Iteration over single ground truth index'''
cls = 5
data = self.image.load()
classes = self.gt.ravel()
pixels = data.reshape((-1, data.shape[-1]))
sum = np.sum(pixels[classes == cls], 0)
itsum = np.sum(np.array([x for x in iterator(data, self.gt, cls)]), 0)
assert_allclose(sum, itsum)
def test_iterator_ij_nonzero(self):
'''Iteration over all non-background pixels.'''
data = self.image.load()
classes = self.gt.ravel()
pixels = data.reshape((-1, data.shape[-1]))
sum = np.sum(pixels[classes > 0], 0)
itsum = np.sum(np.array([data[ij] for ij in iterator_ij(self.gt)]), 0)
assert_allclose(sum, itsum)
def test_iterator_ij_index(self):
'''Iteration over single ground truth index'''
cls = 5
data = self.image.load()
classes = self.gt.ravel()
pixels = data.reshape((-1, data.shape[-1]))
sum = np.sum(pixels[classes == cls], 0)
itsum = np.sum(np.array([data[ij] for ij in iterator_ij(self.gt,
cls)]),
0)
assert_allclose(sum, itsum)
def test_iterator_spyfile(self):
'''Iteration over SpyFile object for single ground truth index'''
cls = 5
data = self.image.load()
classes = self.gt.ravel()
pixels = data.reshape((-1, data.shape[-1]))
sum = np.sum(pixels[classes == cls], 0)
itsum = np.sum(np.array([x for x in iterator(self.image, self.gt, cls)]),
0)
assert_allclose(sum, itsum)
def test_iterator_spyfile_nomemmap(self):
'''Iteration over SpyFile object without memmap'''
cls = 5
data = self.image.load()
classes = self.gt.ravel()
pixels = data.reshape((-1, data.shape[-1]))
sum = np.sum(pixels[classes == cls], 0)
image = spy.open_image('92AV3C.lan')
itsum = np.sum(np.array([x for x in iterator(image, self.gt, cls)]), 0)
assert_allclose(sum, itsum)
def run():
print('\n' + '-' * 72)
print('Running iterator tests.')
print('-' * 72)
test = IteratorTest()
test.run()
if __name__ == '__main__':
from spectral.tests.run import parse_args, reset_stats, print_summary
parse_args()
reset_stats()
run()
print_summary()
|
