diff --git a/tests/unit_tests/test_transform.py b/tests/unit_tests/test_transform.py
new file mode 100644
index 0000000..aff7130
--- /dev/null
+++ b/tests/unit_tests/test_transform.py
@@ -0,0 +1,17 @@
+import numpy as np
+
+from qusi.internal.light_curve import LightCurve, remove_infinite_flux_data_points_from_light_curve
+
+
+def test_remove_infinite_flux_data_points_from_light_curve():
+    times = np.array([0.0, 1.0, 2.0])
+    fluxes = np.array([0.0, np.inf, 20.0])
+    light_curve = LightCurve.new(
+        times=times,
+        fluxes=fluxes,
+    )
+    updated_light_curve = remove_infinite_flux_data_points_from_light_curve(light_curve=light_curve)
+    expected_times = np.array([0.0, 2.0])
+    expected_fluxes = np.array([0.0, 20.0])
+    assert np.array_equal(updated_light_curve.times, expected_times)
+    assert np.array_equal(updated_light_curve.fluxes, expected_fluxes)