Skip to content

Commit 7e2ab29

Browse files
committed
Use fake data in test
1 parent 9874cc5 commit 7e2ab29

File tree

1 file changed

+12
-14
lines changed

1 file changed

+12
-14
lines changed

tests/dream/geant4_reduction_test.py

Lines changed: 12 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -191,24 +191,22 @@ def test_pipeline_can_compute_dspacing_result_with_integrated_monitor_norm(
191191
def test_pipeline_normalizes_and_subtracts_empty_can_as_expected(
192192
workflow: sciline.Pipeline,
193193
) -> None:
194+
sample = sc.data.binned_x(13, 3)
195+
vanadium = sc.data.binned_x(16, 3)
196+
empty_can = sc.data.binned_x(9, 3)
197+
198+
workflow[FocussedDataDspacing[SampleRun]] = sample
199+
workflow[FocussedDataDspacing[VanadiumRun]] = vanadium
200+
workflow[FocussedDataDspacing[BackgroundRun]] = empty_can
194201
workflow[UncertaintyBroadcastMode] = UncertaintyBroadcastMode.drop
195202
workflow = powder.with_pixel_mask_filenames(workflow, [])
196-
results = workflow.compute(
197-
[
198-
EmptyCanSubtractedIofDspacing[SampleRun],
199-
FocussedDataDspacing[SampleRun],
200-
FocussedDataDspacing[VanadiumRun],
201-
FocussedDataDspacing[BackgroundRun],
202-
]
203-
)
204-
result = results[EmptyCanSubtractedIofDspacing[SampleRun]]
205-
206-
sample = results[FocussedDataDspacing[SampleRun]]
207-
empty_can = results[FocussedDataDspacing[BackgroundRun]]
208-
vanadium = results[FocussedDataDspacing[VanadiumRun]]
203+
result = workflow.compute(EmptyCanSubtractedIofDspacing[SampleRun])
209204

205+
subtracted = sample.bins.concatenate(-empty_can)
210206
expected = powder.correction.normalize_by_vanadium_dspacing(
211-
sample.bins.concatenate(-empty_can), vanadium, UncertaintyBroadcastMode.drop
207+
FocussedDataDspacing[SampleRun](subtracted),
208+
FocussedDataDspacing[VanadiumRun](vanadium),
209+
UncertaintyBroadcastMode.drop,
212210
)
213211
sc.testing.assert_allclose(result, expected)
214212

0 commit comments

Comments
 (0)