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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
import pytest
import numpy as np
from PyQt5 import QtWidgets, QtCore
from pyimagetool import ImageTool, RegularDataArray
class TestImageTool:
@staticmethod
def make_regular_data():
# Generate data
mat = np.array([[[9, 1],
[2, 3]],
[[5, 5],
[1, 1]],
[[1, 1],
[0, 0]],
[[3, 2],
[1, 8]]])
delta = [2, 5, 7]
coord_min = [0, 1, 2]
return RegularDataArray(mat, dims=('x4', 'y2', 'z2'), delta=delta, coord_min=coord_min)
@staticmethod
def make_numpy_data():
# Generate data
mat = np.array([[[9, 1],
[2, 3]],
[[5, 5],
[1, 1]],
[[1, 1],
[0, 0]],
[[3, 2],
[1, 8]]])
return mat
def test_imagetool_numpy_show(self, qtbot):
mat = self.make_numpy_data()
it = ImageTool(mat)
qtbot.addWidget(it)
assert len(it.info_bar.cursor_i) == 3
it.show()
def test_imagetool_regular_show(self, qtbot):
dat = self.make_regular_data()
it = ImageTool(dat)
qtbot.addWidget(it)
assert len(it.info_bar.cursor_i) == 3
it.show()
def test_imagetool_numpy(self, qtbot):
mat = self.make_numpy_data()
it = ImageTool(mat)
assert it.pg_win.lineplots_data['x'][1] == 'h'
assert np.all(it.pg_win.lineplots_data['x'][0].xData == np.arange(4))
assert np.all(it.pg_win.lineplots_data['x'][0].yData == mat[:, 0, 0])
assert it.pg_win.lineplots_data['y'][1] == 'v'
assert np.all(it.pg_win.lineplots_data['y'][0].xData == mat[0, :, 0])
assert np.all(it.pg_win.lineplots_data['y'][0].yData == np.arange(2))
assert it.pg_win.lineplots_data['z'][1] == 'v'
assert np.all(it.pg_win.lineplots_data['z'][0].xData == mat[0, 0, :])
assert np.all(it.pg_win.lineplots_data['z'][0].yData == np.arange(2))
def test_imagetool_regular(self, qtbot):
dat = self.make_regular_data()
it = ImageTool(dat)
assert it.pg_win.lineplots_data['x'][1] == 'h'
assert np.all(it.pg_win.lineplots_data['x'][0].xData == dat.axes[0])
assert np.all(it.pg_win.lineplots_data['x'][0].yData == dat.values[:, 0, 0])
assert it.pg_win.lineplots_data['y'][1] == 'v'
assert np.all(it.pg_win.lineplots_data['y'][0].xData == dat.values[0, :, 0])
assert np.all(it.pg_win.lineplots_data['y'][0].yData == dat.axes[1])
assert it.pg_win.lineplots_data['z'][1] == 'v'
assert np.all(it.pg_win.lineplots_data['z'][0].xData == dat.values[0, 0, :])
assert np.all(it.pg_win.lineplots_data['z'][0].yData == dat.axes[2])
def test_imagetool_cursor(self, qtbot):
dat = self.make_regular_data()
it = ImageTool(dat)
it.info_bar.cursor_i[0].setValue(2)
it.info_bar.cursor_i[1].setValue(1)
np.testing.assert_almost_equal(it.pg_win.lineplots_data['x'][0].yData, dat.values[:, 1, 0])
np.testing.assert_almost_equal(it.pg_win.lineplots_data['y'][0].xData, dat.values[2, :, 0])
np.testing.assert_almost_equal(it.pg_win.lineplots_data['z'][0].xData, dat.values[2, 1, :])
it.info_bar.cursor_i[0].setValue(1)
np.testing.assert_almost_equal(it.pg_win.lineplots_data['x'][0].yData, dat.values[:, 1, 0])
np.testing.assert_almost_equal(it.pg_win.lineplots_data['y'][0].xData, dat.values[1, :, 0])
np.testing.assert_almost_equal(it.pg_win.lineplots_data['z'][0].xData, dat.values[1, 1, :])
def test_imagetool_bin(self, qtbot):
dat = self.make_regular_data()
it = ImageTool(dat)
# Set cursor index to (2, 1, 0), and bin in index by 2, so we average over three elements in x
it.info_bar.cursor_i[0].setValue(2)
it.info_bar.cursor_i[1].setValue(1)
it.info_bar.bin_i[0].setValue(2)
qtbot.waitSignal(it.pg_win.cursor._binwidth[0].value_set, timeout=3000)
assert it.info_bar.bin_c[0].value() == pytest.approx(2*dat.delta[0])
np.testing.assert_almost_equal(it.pg_win.lineplots_data['x'][0].yData, dat.values[:, 1, 0])
np.testing.assert_almost_equal(it.pg_win.lineplots_data['y'][0].xData,
np.mean(dat.values[1:4, :, 0:1], axis=(0, 2)))
np.testing.assert_almost_equal(it.pg_win.lineplots_data['z'][0].xData,
np.mean(dat.values[1:4, 1:2, :], axis=(0, 1)))
# Confirm that updating the bin width double spinbox property updates the bin width integer spinbox
it.info_bar.bin_c[0].setValue(4.1)
it.info_bar.bin_c[0].editingFinished.emit()
assert it.info_bar.bin_i[0].value() == 2
# Place the cursor at x = 3, and the index of x = 1.5 => 2
# Set the bin width index to 1. This guarantees no binning.
it.info_bar.cursor_c[0].setValue(3)
it.info_bar.cursor_c[0].editingFinished.emit()
it.info_bar.bin_i[0].setValue(1)
np.testing.assert_almost_equal(it.pg_win.lineplots_data['x'][0].yData, dat.values[:, 1, 0])
np.testing.assert_almost_equal(it.pg_win.lineplots_data['y'][0].xData,
np.mean(dat.values[2:3, :, 0:1], axis=(0, 2)))
np.testing.assert_almost_equal(it.pg_win.lineplots_data['z'][0].xData,
np.mean(dat.values[2:3, 1:2, :], axis=(0, 1)))
# Cursor still at 3. Now the bin width is slightly larger than delta, and there will be averaging.
it.info_bar.bin_c[0].setValue(2.1)
it.info_bar.bin_c[0].editingFinished.emit()
np.testing.assert_almost_equal(it.pg_win.lineplots_data['x'][0].yData, dat.values[:, 1, 0])
np.testing.assert_almost_equal(it.pg_win.lineplots_data['y'][0].xData,
np.mean(dat.values[1:3, :, 0:1], axis=(0, 2)))
np.testing.assert_almost_equal(it.pg_win.lineplots_data['z'][0].xData,
np.mean(dat.values[1:3, 1:2, :], axis=(0, 1)))
def test_imagetool_transpose(self, qtbot):
dat = self.make_regular_data()
it = ImageTool(dat)
qtbot.addWidget(it)
it.info_bar.cursor_i[0].setValue(2)
it.info_bar.cursor_i[1].setValue(1)
it.info_bar.bin_i[0].setValue(2)
dat = dat.transpose([1, 0, 2])
it.info_bar.transpose_request.emit([1, 0, 2])
# dat = dat.transpose({0: 1, 1: 0, 2: 2})
# it.info_bar.transpose_request.emit({0: 1, 1: 0, 2: 2})
assert it.info_bar.cursor_labels[0].text() == 'y2'
assert it.info_bar.cursor_labels[1].text() == 'x4'
assert it.info_bar.cursor_labels[2].text() == 'z2'
it.info_bar.bin_i[1].setValue(2)
np.testing.assert_almost_equal(it.pg_win.lineplots_data['x'][0].xData, dat.axes[0])
np.testing.assert_almost_equal(it.pg_win.lineplots_data['y'][0].yData, dat.axes[1])
np.testing.assert_almost_equal(it.pg_win.lineplots_data['z'][0].yData, dat.axes[2])
np.testing.assert_almost_equal(it.pg_win.lineplots_data['x'][0].yData,
np.mean(dat.values[:, 0:2, 0:1], axis=(1, 2)))
np.testing.assert_almost_equal(it.pg_win.lineplots_data['y'][0].xData, dat.values[0, :, 0])
np.testing.assert_almost_equal(it.pg_win.lineplots_data['z'][0].xData,
np.mean(dat.values[0:1, 0:2, :], axis=(0, 1)))
|
