Skip to content
Snippets Groups Projects
Commit 0d6e332b authored by Bert Palm's avatar Bert Palm 🎇
Browse files

fixed iloc bugs in tests

parent f8950ee6
No related branches found
No related tags found
4 merge requests!193Release 1.4,!188Release 1.4,!32WIP: Plot rework,!24Dios integration
Pipeline #3170 failed with stage
in 7 minutes and 3 seconds
......@@ -135,15 +135,16 @@ def test_flagIsolated(data, flagger):
data.iloc[11:13, 0] = np.nan
data.iloc[15:17, 0] = np.nan
flagger = flagger.initFlags(data)
flagger = flagger.setFlags(field, iloc=slice(5, 6))
s = data[field].iloc[5:6]
flagger = flagger.setFlags(field, loc=s)
_, flagger_result = flagIsolated(data, field, flagger, group_window="1D", gap_window="2.1D")
assert flagger_result.isFlagged(field)[slice(3, 6, 2)].all()
flagger = flagger.setFlags(
field, iloc=slice(3, 4), flag=flagger.UNFLAGGED, force=True
)
# flagger = flagger.setFlags(
# field, iloc=slice(3, 4), flag=flagger.UNFLAGGED, force=True
# )
data, flagger_result = flagIsolated(
data,
field,
......
......@@ -201,8 +201,8 @@ def test_isflagged(data, flagger):
flagger = flagger.initFlags(data)
var1, var2, *_ = data.columns
flagger = flagger.setFlags(var1, iloc=slice(None, None, 2))
flagger = flagger.setFlags(var2, iloc=slice(None, None, 2))
flagger = flagger.setFlags(var1, loc=slice(None, None, 2))
flagger = flagger.setFlags(var2, loc=slice(None, None, 2))
idx = _evalDslExpression(f"isflagged({var1})", data, var2, flagger)
......@@ -216,7 +216,7 @@ def test_invertIsFlagged(data, flagger):
flagger = flagger.initFlags(data)
var1, var2, *_ = data.columns
flagger = flagger.setFlags(var2, iloc=slice(None, None, 2))
flagger = flagger.setFlags(var2, loc=slice(None, None, 2))
tests = [
(f"~isflagged({var2})", ~flagger.isFlagged(var2)),
......@@ -237,7 +237,7 @@ def test_isflaggedArgument(data, flagger):
var1, var2, *_ = data.columns
flagger = flagger.initFlags(data).setFlags(
var1, iloc=slice(None, None, 2), flag=flagger.BAD
var1, loc=slice(None, None, 2), flag=flagger.BAD
)
tests = [
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment