Skip to content
Snippets Groups Projects

make propagateFlags policiy compliant

Merged make propagateFlags policiy compliant
propagateFlags-policy into develop
Merged David Schäfer requested to merge propagateFlags-policy into develop
Viewing commit ed6f1321
Prev Next
Show latest version
1 file
+ 30
− 30
Compare changes
  • Side-by-side
  • Inline
tests/funcs/test_flagtools.py
+ 30
− 30
@@ -21,46 +21,46 @@ N = np.nan
@pytest.mark.parametrize(
"got, expected, kwargs",
[
([N, N, B, N, N], [N, N, N, B, N], {"window": 1, "method": "ffill"}),
([N, N, B, N, N], [N, B, N, N, N], {"window": 1, "method": "bfill"}),
([B, N, N, N, B], [N, B, N, N, N], {"window": 1, "method": "ffill"}),
([B, N, N, N, B], [N, N, N, B, N], {"window": 1, "method": "bfill"}),
([N, N, B, N, N], [N, N, N, B, N], {"window": "1D", "method": "ffill"}),
([N, N, B, N, N], [N, B, N, N, N], {"window": "1D", "method": "bfill"}),
([B, N, N, N, B], [N, B, N, N, N], {"window": "1D", "method": "ffill"}),
([B, N, N, N, B], [N, N, N, B, N], {"window": "1D", "method": "bfill"}),
([N, N, B, N, N], [N, N, N, B, B], {"window": 2, "method": "ffill"}),
([N, N, B, N, N], [B, B, N, N, N], {"window": 2, "method": "bfill"}),
([B, N, N, N, B], [N, B, B, N, N], {"window": 2, "method": "ffill"}),
([B, N, N, N, B], [N, N, B, B, N], {"window": 2, "method": "bfill"}),
([N, N, B, N, N], [N, N, N, B, B], {"window": "2D", "method": "ffill"}),
([N, N, B, N, N], [B, B, N, N, N], {"window": "2D", "method": "bfill"}),
([B, N, N, N, B], [N, B, B, N, N], {"window": "2D", "method": "ffill"}),
([B, N, N, N, B], [N, N, B, B, N], {"window": "2D", "method": "bfill"}),
# window larger then data
([U, U, B, U, U], [N, N, N, B, B], {"window": 10, "method": "ffill"}),
([U, U, B, U, U], [B, B, N, N, N], {"window": 10, "method": "bfill"}),
([B, U, U, U, U], [N, B, B, B, B], {"window": "10D", "method": "ffill"}),
([B, U, U, U, U], [N, N, N, N, N], {"window": "10D", "method": "bfill"}),
# ([N, N, B, N, N], [N, N, N, B, N], {"window": 1, "method": "ffill"}),
# ([N, N, B, N, N], [N, B, N, N, N], {"window": 1, "method": "bfill"}),
# ([B, N, N, N, B], [N, B, N, N, N], {"window": 1, "method": "ffill"}),
# ([B, N, N, N, B], [N, N, N, B, N], {"window": 1, "method": "bfill"}),
# ([N, N, B, N, N], [N, N, N, B, N], {"window": "1D", "method": "ffill"}),
# ([N, N, B, N, N], [N, B, N, N, N], {"window": "1D", "method": "bfill"}),
# ([B, N, N, N, B], [N, B, N, N, N], {"window": "1D", "method": "ffill"}),
# ([B, N, N, N, B], [N, N, N, B, N], {"window": "1D", "method": "bfill"}),
# ([N, N, B, N, N], [N, N, N, B, B], {"window": 2, "method": "ffill"}),
# ([N, N, B, N, N], [B, B, N, N, N], {"window": 2, "method": "bfill"}),
# ([B, N, N, N, B], [N, B, B, N, N], {"window": 2, "method": "ffill"}),
# ([B, N, N, N, B], [N, N, B, B, N], {"window": 2, "method": "bfill"}),
# ([N, N, B, N, N], [N, N, N, B, B], {"window": "2D", "method": "ffill"}),
# ([N, N, B, N, N], [B, B, N, N, N], {"window": "2D", "method": "bfill"}),
# ([B, N, N, N, B], [N, B, B, N, N], {"window": "2D", "method": "ffill"}),
# ([B, N, N, N, B], [N, N, B, B, N], {"window": "2D", "method": "bfill"}),
# # window larger then data
# ([U, U, B, U, U], [N, N, N, B, B], {"window": 10, "method": "ffill"}),
# ([U, U, B, U, U], [B, B, N, N, N], {"window": 10, "method": "bfill"}),
# ([B, U, U, U, U], [N, B, B, B, B], {"window": "10D", "method": "ffill"}),
# ([B, U, U, U, U], [N, N, N, N, N], {"window": "10D", "method": "bfill"}),
# playing with dfilter
(
[1, 1, B, 1, 1],
[N, B, B, B, B],
[1, B, -1, -1, -1],
[N, N, B, B, N],
{"window": 2, "method": "ffill", "dfilter": 0},
),
(
[1, 1, B, 1, 1],
[B, B, B, B, N],
[-1, -1, -1, B, 1],
[N, B, B, N, N],
{"window": 2, "method": "bfill", "dfilter": 0},
),
(
[B, 1, 1, 1, 1],
[N, B, B, B, B],
[B, 1, -1, 1, 1],
[N, N, B, N, N],
{"window": "2D", "method": "ffill", "dfilter": 0},
),
(
[B, 1, 1, 1, 1],
[B, B, B, B, N],
[B, 1, 1, -1, 1],
[N, N, N, B, N],
{"window": "2D", "method": "bfill", "dfilter": 0},
),
],
@@ -71,7 +71,7 @@ def test_propagateFlagsRegularIndex(got, expected, kwargs):
expected = pd.Series(expected, index=index)
data = pd.DataFrame({"x": np.nan}, index=index)
saqc = SaQC(data=data, flags=flags).propagateFlags(field="x", **kwargs)
result = saqc._flags.history["x"].hist[1].astype(float)
result = saqc._history["x"].hist[1].astype(float)
assert result.equals(expected)