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 unittest.mock
import io
import numpy as np
import matplotlib
matplotlib.use('Agg') # Set the backend to Agg
from mcstasscript.jb_interface.plot_interface import PlotInterface
from mcstasscript.jb_interface.plot_interface import LogCheckbox
from mcstasscript.jb_interface.plot_interface import ColormapDropdown
from mcstasscript.jb_interface.plot_interface import OrdersOfMagField
from mcstasscript.data.data import McStasMetaData
from mcstasscript.data.data import McStasDataBinned
import ipywidgets as widgets
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
meta_data.limits = [0, 1]
meta_data.xlabel = ""
meta_data.ylabel = ""
meta_data.title = ""
meta_data.filename = "dummy"
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 = [5, 4]
meta_data.limits = [0, 1, 0, 1]
meta_data.xlabel = ""
meta_data.ylabel = ""
meta_data.title = ""
meta_data.filename = "dummy"
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)
def fake_data():
return [set_dummy_McStasDataBinned_1d(),
set_dummy_McStasDataBinned_2d(),
set_dummy_McStasDataBinned_2d()]
class FakeChange:
def __init__(self, new=None, old=None, name=None):
self.new = new
self.old = old
self.name = name
class TestPlotInterface(unittest.TestCase):
"""
Test of PlotInterface, mainly the set functions.
Each set function does run the main update_plot function, ensuring
this does not throw obvious errors under a mixed set of circumstances.
The update_plot function is not tested directly.
"""
def test_initialization_without_data(self):
"""
Initialize PlotterInterface without arguments
"""
interface = PlotInterface()
self.assertEqual(interface.data, None)
def test_initialization_with_data(self):
"""
Initialize PlotterInterface with data
"""
data = fake_data()
interface = PlotInterface(data)
self.assertEqual(len(interface.data), 3)
self.assertEqual(interface.data[0].Intensity[5], 5)
self.assertEqual(interface.data[1].Intensity[2, 3], 13)
def test_show_interface_return(self):
"""
Ensure the show_interface method returns a widget of type HBox
"""
interface = PlotInterface()
widget = interface.show_interface()
self.assertIsInstance(widget, widgets.widgets.widget_box.HBox)
def test_set_data(self):
"""
Initialize PlotterInterface without data, add later and ensure
monitor_dropdown received the data and applied it to widget options.
Furthermore the widget options should be made unique as there are
two identical monitors in the fake data.
"""
interface = PlotInterface()
self.assertEqual(interface.data, None)
interface.show_interface() # Needed to set up monitor_dropdown
interface.set_data(fake_data())
self.assertEqual(len(interface.data), 3)
self.assertEqual(interface.data[0].Intensity[5], 5)
self.assertEqual(interface.data[1].Intensity[2, 3], 13)
mon_drop = interface.monitor_dropdown
self.assertEqual(len(mon_drop.data), 3)
self.assertEqual(mon_drop.data[0].Intensity[5], 5)
self.assertEqual(mon_drop.data[1].Intensity[2, 3], 13)
self.assertEqual(mon_drop.widget.options[0], "component for 1d")
self.assertEqual(mon_drop.widget.options[1], "test a component")
# The last monitor is identical to the second, so the name is modified
self.assertEqual(mon_drop.widget.options[2], "test a component_1")
def test_set_current_monitor(self):
"""
Check monitor can be set and that error occurs if wrong name given
"""
interface = PlotInterface(fake_data())
interface.show_interface()
interface.set_current_monitor("test a component")
self.assertEqual(interface.current_monitor, "test a component")
with self.assertRaises(NameError):
interface.set_current_monitor("component that doesnt exist")
def test_set_log_mode(self):
"""
Check that set_log_mode works, even through widget
"""
interface = PlotInterface()
interface.show_interface()
self.assertFalse(interface.log_mode)
interface.set_log_mode(True)
self.assertTrue(interface.log_mode)
log_checkbox = LogCheckbox(interface.log_mode, interface.set_log_mode)
fake_change = FakeChange(new=False)
log_checkbox.update(fake_change)
self.assertFalse(interface.log_mode)
def test_set_orders_of_mag(self):
"""
Check that set orders of mag works, even through widget
"""
interface = PlotInterface()
interface.show_interface()
interface.set_orders_of_mag(37)
self.assertEqual(interface.orders_of_mag, 37)
log_orders_of_mag = OrdersOfMagField(interface.set_orders_of_mag)
fake_change = FakeChange(new=42)
log_orders_of_mag.update(fake_change)
self.assertEqual(interface.orders_of_mag, 42)
def test_set_colormap(self):
"""
Check that set_colormap works, even through widget
"""
interface = PlotInterface()
interface.show_interface()
interface.set_colormap("hot")
self.assertEqual(interface.colormap, "hot")
colormap_dropdown = ColormapDropdown(interface.set_colormap)
fake_change = FakeChange(new="Purples")
colormap_dropdown.update_cmap(fake_change)
self.assertEqual(interface.colormap, "Purples")
|
