improve coverage

tests/node_tests/line_correction.py

@@ -52,3 +52,87 @@ def test_line_correction():
     assert np.allclose(leveled.data + poly_bg.data, poly_field.data)
     assert np.corrcoef(leveled.data.ravel(), poly_signal.ravel())[0, 1] > 0.995
     assert len(poly_shifts) == rows
+
+
+def test_line_correction_methods():
+    from backend.nodes.line_correction import LineCorrection
+    from tests.node_tests._shared import make_field
+
+    node = LineCorrection()
+
+    rows, cols = 64, 80
+    rng = np.random.default_rng(7)
+    signal = rng.standard_normal((rows, cols)) * 0.1
+    row_offsets = rng.standard_normal(rows) * 2.0
+    data = signal + row_offsets[:, None]
+    field = make_field(data=data)
+
+    # median_diff
+    c, b, s = node.process(field, method="median_diff", direction="horizontal",
+                           masking="ignore", trim_fraction=0.05, polynomial_degree=1)
+    assert np.allclose(c.data + b.data, field.data)
+    assert len(s) == rows
+
+    # trimmed_mean
+    c, b, s = node.process(field, method="trimmed_mean", direction="horizontal",
+                           masking="ignore", trim_fraction=0.2, polynomial_degree=1)
+    assert np.allclose(c.data + b.data, field.data)
+
+    # trimmed_diff
+    c, b, s = node.process(field, method="trimmed_diff", direction="horizontal",
+                           masking="ignore", trim_fraction=0.2, polynomial_degree=1)
+    assert np.allclose(c.data + b.data, field.data)
+
+    # step
+    c, b, s = node.process(field, method="step", direction="horizontal",
+                           masking="ignore", trim_fraction=0.05, polynomial_degree=1)
+    assert np.allclose(c.data + b.data, field.data)
+    assert len(s) == rows
+
+
+def test_line_correction_vertical():
+    from backend.nodes.line_correction import LineCorrection
+    from tests.node_tests._shared import make_field
+
+    node = LineCorrection()
+
+    rows, cols = 48, 64
+    col_offsets = np.random.default_rng(3).standard_normal(cols) * 1.5
+    data = np.random.default_rng(3).standard_normal((rows, cols)) * 0.1 + col_offsets[None, :]
+    field = make_field(data=data)
+
+    c, b, s = node.process(field, method="median", direction="vertical",
+                           masking="ignore", trim_fraction=0.05, polynomial_degree=1)
+    assert c.data.shape == field.data.shape
+    assert np.allclose(c.data + b.data, field.data)
+    # vertical shift line length = number of columns
+    assert len(s) == cols
+    assert s.x_axis is not None
+    assert np.isclose(s.x_axis[-1], field.xreal)
+
+
+def test_line_correction_with_mask():
+    from backend.nodes.line_correction import LineCorrection
+    from tests.node_tests._shared import make_field
+
+    node = LineCorrection()
+
+    rows, cols = 32, 48
+    data = np.random.default_rng(9).standard_normal((rows, cols)) * 0.1
+    row_offsets = np.linspace(0, 3.0, rows)
+    data += row_offsets[:, None]
+    field = make_field(data=data)
+
+    # mask covers right half
+    mask = np.zeros((rows, cols), dtype=np.uint8)
+    mask[:, cols // 2:] = 255
+
+    c_excl, b_excl, _ = node.process(field, method="median", direction="horizontal",
+                                      masking="exclude", trim_fraction=0.05,
+                                      polynomial_degree=1, mask=mask)
+    assert np.allclose(c_excl.data + b_excl.data, field.data)
+
+    c_incl, b_incl, _ = node.process(field, method="median", direction="horizontal",
+                                      masking="include", trim_fraction=0.05,
+                                      polynomial_degree=1, mask=mask)
+    assert np.allclose(c_incl.data + b_incl.data, field.data)