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

extended assign test

parent 626b5977
No related branches found
No related tags found
No related merge requests found
...@@ -95,19 +95,22 @@ def test_assignVariable(flagger): ...@@ -95,19 +95,22 @@ def test_assignVariable(flagger):
meta = prepareMeta( meta = prepareMeta(
pd.DataFrame( pd.DataFrame(
{Fields.VARNAME: [var1, var2], {Fields.VARNAME: [var1, var2],
Fields.FLAGS: ["range, {min: -9999, max: 9999}", Fields.FLAGS: ["range, {min: 99999, max: -99999}",
f"generic, {{func: isflagged({var2}), assign: True}}"]}), f"generic, {{func: isflagged({var1}), assign: True}}"]}),
data) data)
pdata, pflags = runner(meta, flagger, data) pdata, pflags = runner(meta, flagger, data)
if isinstance(pflags.columns, pd.MultiIndex): if isinstance(pflags.columns, pd.MultiIndex):
cols = (pflags cols = (pflags
.columns.get_level_values(0) .columns.get_level_values(0)
.drop_duplicates()) .drop_duplicates())
assert (cols == [var1, var2]).all() assert (cols == [var1, var2]).all()
assert flagger.isFlagged(pflags[var2]).any()
else: else:
assert (pflags.columns == [var1, var2]).all() assert (pflags.columns == [var1, var2]).all()
assert flagger.isFlagged(pflags[var2]).any()
@pytest.mark.parametrize("flagger", TESTFLAGGERS) @pytest.mark.parametrize("flagger", TESTFLAGGERS)
......
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