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
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
|
import unittest
import numpy as np
from mcstasscript.data.data import McStasData
from mcstasscript.data.data import McStasDataBinned
from mcstasscript.data.data import McStasDataEvent
from mcstasscript.data.data import McStasMetaData
def set_dummy_MetaDataBinned_1d():
"""
Sets up simple McStasMetaData object with dimension, 1d case
"""
meta_data = McStasMetaData()
meta_data.component_name = "component for 1d"
meta_data.dimension = 50
return meta_data
def set_dummy_McStasDataBinned_1d():
"""
Sets up simple McStasData object, 1d case
"""
meta_data = set_dummy_MetaDataBinned_1d()
intensity = np.arange(20)
error = 0.5 * np.arange(20)
ncount = 2 * np.arange(20)
axis = np.arange(20)*5.0
return McStasDataBinned(meta_data, intensity, error, ncount, xaxis=axis)
def set_dummy_MetaDataBinned_2d():
"""
Sets up simple McStasMetaData object with dimensions, 2d case
"""
meta_data = McStasMetaData()
meta_data.component_name = "test a component"
meta_data.dimension = [50, 100]
return meta_data
def set_dummy_McStasDataBinned_2d():
"""
Sets up simple McStasData object, 2d case
"""
meta_data = set_dummy_MetaDataBinned_2d()
intensity = np.arange(20).reshape(4, 5)
error = 0.5 * np.arange(20).reshape(4, 5)
ncount = 2 * np.arange(20).reshape(4, 5)
return McStasDataBinned(meta_data, intensity, error, ncount)
class TestMcStasData(unittest.TestCase):
"""
Various tests of McStasData class
"""
def test_McStasDataBinned_init_1d(self):
"""
Test that newly created McStasMetaData has correct names, 1d case
"""
data = set_dummy_McStasDataBinned_1d()
self.assertEqual(data.name, "component for 1d")
self.assertEqual(data.metadata.component_name, "component for 1d")
def test_McStasDataBinned_init_values(self):
"""
Test that newly created McStasDataBinned has expected data, 1d case
Here checking a single data point
"""
data = set_dummy_McStasDataBinned_1d()
self.assertEqual(data.Intensity[3], 3)
self.assertEqual(data.Error[3], 1.5)
self.assertEqual(data.Ncount[3], 6)
self.assertEqual(data.xaxis[3], 15.0)
def test_McStasDataBinned_init_values_full(self):
"""
Test that newly created McStasDataBinned has expected data, 1d case
"""
data = set_dummy_McStasDataBinned_1d()
intensity = np.arange(20)
error = 0.5 * np.arange(20)
ncount = 2 * np.arange(20)
axis = np.arange(20) * 5.0
for index in range(len(data.Intensity)):
self.assertEqual(data.Intensity[index], intensity[index])
self.assertEqual(data.Error[index], error[index])
self.assertEqual(data.Ncount[index], ncount[index])
self.assertEqual(data.xaxis[index], axis[index])
def test_McStasDataBinned_init_2d_names(self):
"""
Test that newly created McStasMetaData has correct names, 1d case
"""
data = set_dummy_McStasDataBinned_2d()
self.assertEqual(data.name, "test a component")
self.assertEqual(data.metadata.component_name, "test a component")
def test_McStasDataBinned_init_2d_values(self):
"""
Test that newly created McStasDataBinned has expected data, 2d case
Here checking a single point
"""
data = set_dummy_McStasDataBinned_2d()
self.assertEqual(data.Intensity[2][3], 13)
self.assertEqual(data.Error[2][3], 6.5)
self.assertEqual(data.Ncount[2][3], 26)
def test_McStasDataBinned_init_2d_values_full(self):
"""
Test that newly created McStasDataBinned has expected data, 2d case
Here checking a entire dataset
"""
data = set_dummy_McStasDataBinned_2d()
intensity = np.arange(20).reshape(4, 5)
error = 0.5 * np.arange(20).reshape(4, 5)
ncount = 2 * np.arange(20).reshape(4, 5)
shape = np.shape(data.Intensity)
for index1 in range(shape[0]):
for index2 in range(shape[1]):
self.assertEqual(data.Intensity[index1][index2],
intensity[index1][index2])
self.assertEqual(data.Error[index1][index2],
error[index1][index2])
self.assertEqual(data.Ncount[index1][index2],
ncount[index1][index2])
def test_McStasDataBinned_set_info_title(self):
"""
Test that title can be set
"""
data = set_dummy_McStasDataBinned_2d()
data.set_title("title_test")
self.assertEqual(data.metadata.title, "title_test")
def test_McStasDataBinned_set_xlabel(self):
"""
Test that xlabel can be set
"""
data = set_dummy_McStasDataBinned_2d()
data.set_xlabel("xlabel test")
self.assertEqual(data.metadata.xlabel, "xlabel test")
def test_McStasDataBinned_set_ylabel(self):
"""
Test that ylabel can be set
"""
data = set_dummy_McStasDataBinned_2d()
data.set_ylabel("ylabel test")
self.assertEqual(data.metadata.ylabel, "ylabel test")
def test_McStasDataBinned_set_log(self):
"""
Test that log setting has correct type regardless of how it is given
"""
data = set_dummy_McStasDataBinned_2d()
data.set_plot_options(log=True)
self.assertIsInstance(data.plot_options.log, bool)
self.assertTrue(data.plot_options.log)
data.set_plot_options(log=0)
self.assertIsInstance(data.plot_options.log, bool)
self.assertFalse(data.plot_options.log)
data.set_plot_options(log=1)
self.assertIsInstance(data.plot_options.log, bool)
self.assertTrue(data.plot_options.log)
def test_McStasDataBinned_set_show_colorbar(self):
"""
Test that log setting has correct type regardless of how it is given
"""
data = set_dummy_McStasDataBinned_2d()
data.set_plot_options(show_colorbar=True)
self.assertIsInstance(data.plot_options.show_colorbar, bool)
self.assertTrue(data.plot_options.show_colorbar)
data.set_plot_options(show_colorbar=0)
self.assertIsInstance(data.plot_options.show_colorbar, bool)
self.assertFalse(data.plot_options.show_colorbar)
data.set_plot_options(show_colorbar=1)
self.assertIsInstance(data.plot_options.show_colorbar, bool)
self.assertTrue(data.plot_options.show_colorbar)
def test_McStasDataBinned_set_orders_of_mag(self):
"""
Test that orders_og_mag can be set correctly
"""
data = set_dummy_McStasDataBinned_2d()
data.set_plot_options(orders_of_mag=5.2)
self.assertEqual(data.plot_options.orders_of_mag, 5.2)
def test_McStasDataBinned_set_colormap(self):
"""
Test that colormap can be set correctly
"""
data = set_dummy_McStasDataBinned_2d()
data.set_plot_options(colormap="hot")
self.assertIs(data.plot_options.colormap, "hot")
if __name__ == '__main__':
unittest.main()
|
