1111
1212import tof
1313
14- from .common import (
15- dummy_chopper ,
16- dummy_detector ,
17- dummy_source ,
18- make_chopper ,
19- make_source ,
20- )
21-
2214Hz = sc .Unit ('Hz' )
2315deg = sc .Unit ('deg' )
2416meter = sc .Unit ('m' )
2517ms = sc .Unit ('ms' )
2618
2719
28- def test_one_chopper_one_opening ():
20+ def test_one_chopper_one_opening (make_chopper , make_source ):
2921 # Make a chopper open from 10-20 ms. Assume zero phase.
3022 topen = 10.0 * ms
3123 tclose = 20.0 * ms
@@ -79,7 +71,7 @@ def test_one_chopper_one_opening():
7971 )
8072
8173
82- def test_two_choppers_one_opening ():
74+ def test_two_choppers_one_opening (make_chopper , make_source ):
8375 # Make a first chopper open from 5-16 ms. Assume zero phase.
8476 topen = 5.0 * ms
8577 tclose = 16.0 * ms
@@ -166,7 +158,7 @@ def test_two_choppers_one_opening():
166158 )
167159
168160
169- def test_two_choppers_one_and_two_openings ():
161+ def test_two_choppers_one_and_two_openings (make_chopper , make_source ):
170162 topen = 5.0 * ms
171163 tclose = 16.0 * ms
172164 chopper1 = make_chopper (
@@ -224,7 +216,7 @@ def test_two_choppers_one_and_two_openings():
224216 )
225217
226218
227- def test_neutron_conservation ():
219+ def test_neutron_conservation (make_chopper ):
228220 N = 100_000
229221 source = tof .Source (facility = 'ess' , neutrons = N )
230222
@@ -266,20 +258,22 @@ def test_neutron_conservation():
266258 assert det .sum ().value + det .masks ['blocked_by_others' ].sum ().value == N
267259
268260
269- def test_add_chopper_and_detector ():
270- chopper = dummy_chopper ()
271- detector = dummy_detector ()
272- model = tof .Model (source = dummy_source () )
261+ def test_add_chopper_and_detector (dummy_chopper , dummy_detector , dummy_source ):
262+ chopper = dummy_chopper
263+ detector = dummy_detector
264+ model = tof .Model (source = dummy_source )
273265 model .add (chopper )
274266 assert 'dummy_chopper' in model .choppers
275267 model .add (detector )
276268 assert 'dummy_detector' in model .detectors
277269
278270
279- def test_add_components_with_same_name_raises ():
280- chopper = dummy_chopper ()
281- detector = dummy_detector ()
282- model = tof .Model (source = dummy_source ())
271+ def test_add_components_with_same_name_raises (
272+ dummy_chopper , dummy_detector , dummy_source
273+ ):
274+ chopper = dummy_chopper
275+ detector = dummy_detector
276+ model = tof .Model (source = dummy_source )
283277 model .add (chopper )
284278 with pytest .raises (
285279 KeyError , match = 'Component with name dummy_chopper already exists'
@@ -297,76 +291,78 @@ def test_add_components_with_same_name_raises():
297291 model .add (detector2 )
298292
299293
300- def test_create_model_with_duplicate_component_names_raises ():
301- chopper = dummy_chopper ()
302- detector = dummy_detector ()
294+ def test_create_model_with_duplicate_component_names_raises (
295+ dummy_chopper , dummy_detector , dummy_source
296+ ):
297+ chopper = dummy_chopper
298+ detector = dummy_detector
303299 with pytest .raises (
304300 ValueError , match = "More than one component named 'dummy_chopper' found"
305301 ):
306- tof .Model (source = dummy_source () , choppers = [chopper , chopper ])
302+ tof .Model (source = dummy_source , choppers = [chopper , chopper ])
307303 with pytest .raises (
308304 ValueError , match = "More than one component named 'dummy_detector' found"
309305 ):
310- tof .Model (source = dummy_source () , detectors = [detector , detector ])
306+ tof .Model (source = dummy_source , detectors = [detector , detector ])
311307
312308
313- def test_iter ():
314- chopper = dummy_chopper ()
315- detector = dummy_detector ()
316- model = tof .Model (source = dummy_source () )
309+ def test_iter (dummy_chopper , dummy_detector , dummy_source ):
310+ chopper = dummy_chopper
311+ detector = dummy_detector
312+ model = tof .Model (source = dummy_source )
317313 model .add (chopper )
318314 assert 'dummy_chopper' in model .choppers
319315 model .add (detector )
320316 assert 'dummy_detector' in model .detectors
321317
322318
323- def test_remove ():
324- chopper = dummy_chopper ()
325- detector = dummy_detector ()
326- model = tof .Model (source = dummy_source () , choppers = [chopper ], detectors = [detector ])
319+ def test_remove (dummy_chopper , dummy_detector , dummy_source ):
320+ chopper = dummy_chopper
321+ detector = dummy_detector
322+ model = tof .Model (source = dummy_source , choppers = [chopper ], detectors = [detector ])
327323 del model ['dummy_chopper' ]
328324 assert 'dummy_chopper' not in model
329325 assert 'dummy_detector' in model
330326 del model ['dummy_detector' ]
331327 assert 'dummy_detector' not in model
332328
333329
334- def test_getitem ():
335- chopper = dummy_chopper ()
336- detector = dummy_detector ()
337- model = tof .Model (source = dummy_source () , choppers = [chopper ], detectors = [detector ])
330+ def test_getitem (dummy_chopper , dummy_detector , dummy_source ):
331+ chopper = dummy_chopper
332+ detector = dummy_detector
333+ model = tof .Model (source = dummy_source , choppers = [chopper ], detectors = [detector ])
338334 assert model ['dummy_chopper' ] is chopper
339335 assert model ['dummy_detector' ] is detector
340336 with pytest .raises (KeyError , match = 'No component with name foo' ):
341337 model ['foo' ]
342338
343339
344- def test_input_can_be_single_component ():
345- chopper = dummy_chopper ()
346- detector = dummy_detector ()
347- model = tof .Model (source = dummy_source () , choppers = chopper , detectors = detector )
340+ def test_input_can_be_single_component (dummy_chopper , dummy_detector , dummy_source ):
341+ chopper = dummy_chopper
342+ detector = dummy_detector
343+ model = tof .Model (source = dummy_source , choppers = chopper , detectors = detector )
348344 assert 'dummy_chopper' in model .choppers
349345 assert 'dummy_detector' in model .detectors
350346
351347
352- def test_bad_input_type_raises ():
353- chopper = dummy_chopper ()
354- detector = dummy_detector ()
348+ def test_bad_input_type_raises (dummy_chopper , dummy_detector , dummy_source ):
349+ chopper = dummy_chopper
350+ detector = dummy_detector
355351 with pytest .raises (TypeError , match = 'Invalid input type' ):
356- _ = tof .Model (source = dummy_source () , choppers = 'bad chopper' )
352+ _ = tof .Model (source = dummy_source , choppers = 'bad chopper' )
357353 with pytest .raises (TypeError , match = 'Invalid input type' ):
358- _ = tof .Model (source = dummy_source () , choppers = [chopper ], detectors = 'abc' )
354+ _ = tof .Model (source = dummy_source , choppers = [chopper ], detectors = 'abc' )
359355 with pytest .raises (TypeError , match = 'Invalid input type' ):
360- _ = tof .Model (source = dummy_source () , choppers = [chopper , 'bad chopper' ])
356+ _ = tof .Model (source = dummy_source , choppers = [chopper , 'bad chopper' ])
361357 with pytest .raises (TypeError , match = 'Invalid input type' ):
362- _ = tof .Model (source = dummy_source () , detectors = (1234 , detector ))
358+ _ = tof .Model (source = dummy_source , detectors = (1234 , detector ))
363359 with pytest .raises (TypeError , match = 'Invalid input type' ):
364- _ = tof .Model (source = dummy_source () , choppers = [detector ])
360+ _ = tof .Model (source = dummy_source , choppers = [detector ])
365361 with pytest .raises (TypeError , match = 'Invalid input type' ):
366- _ = tof .Model (source = dummy_source () , detectors = [chopper ])
362+ _ = tof .Model (source = dummy_source , detectors = [chopper ])
367363
368364
369- def test_model_repr_does_not_raise ():
365+ def test_model_repr_does_not_raise (make_chopper , make_source ):
370366 N = 10_000
371367 source = tof .Source (facility = 'ess' , neutrons = N )
372368 chopper1 = make_chopper (
@@ -392,7 +388,7 @@ def test_model_repr_does_not_raise():
392388 assert repr (model ) is not None
393389
394390
395- def test_component_distance ():
391+ def test_component_distance (make_chopper , make_source ):
396392 # Make a chopper open from 10-20 ms. Assume zero phase.
397393 topen = 10.0 * ms
398394 tclose = 20.0 * ms
0 commit comments