From 75d7e66883127dc93f306f8d0b532da7bbcd02ab Mon Sep 17 00:00:00 2001
From: Luis Sanchez <45677170+LuisSanchez25@users.noreply.github.com>
Date: Tue, 30 Jul 2024 09:21:28 -0500
Subject: [PATCH] Add test correction (#131)

* Add test correction

* Update test_correction.py
---
 .../corrections/implementations/__init__.py    |  1 +
 .../implementations/test_correction.py         | 18 ++++++++++++++++++
 2 files changed, 19 insertions(+)
 create mode 100644 xedocs/schemas/corrections/implementations/test_correction.py

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