diff --git a/xedocs/schemas/corrections/implementations/__init__.py b/xedocs/schemas/corrections/implementations/__init__.py index 06ca1f1..6df6633 100644 --- a/xedocs/schemas/corrections/implementations/__init__.py +++ b/xedocs/schemas/corrections/implementations/__init__.py @@ -25,3 +25,4 @@ from .z_bias import * from .cs2_aft_corrections import * from .se_gain_partition import * +from .test_correction import * diff --git a/xedocs/schemas/corrections/implementations/test_correction.py b/xedocs/schemas/corrections/implementations/test_correction.py new file mode 100644 index 0000000..a517a1a --- /dev/null +++ b/xedocs/schemas/corrections/implementations/test_correction.py @@ -0,0 +1,18 @@ +# This will be a correction designed for testing purposes. +# This SHOULD NOT be used in production. + +import rframe + +from ..base_corrections import TimeSampledCorrection +from ...constants import DETECTOR + +class TestCorrection(TimeSampledCorrection): + _ALIAS = "test_corrections" + + # Correction identical to PMT gains. + # Used for testing + + detector: DETECTOR = rframe.Index() + pmt: int = rframe.Index(ge=0) + + value: float