diff --git a/test/test_core.py b/test/test_core.py index 20c592b6af683704fdfcd76da01c4e182ad0c2c9..5f6b0c805ac855bbfb9ebe9c864d9e09fef9c229 100644 --- a/test/test_core.py +++ b/test/test_core.py @@ -95,19 +95,22 @@ def test_assignVariable(flagger): meta = prepareMeta( pd.DataFrame( {Fields.VARNAME: [var1, var2], - Fields.FLAGS: ["range, {min: -9999, max: 9999}", - f"generic, {{func: isflagged({var2}), assign: True}}"]}), + Fields.FLAGS: ["range, {min: 99999, max: -99999}", + f"generic, {{func: isflagged({var1}), assign: True}}"]}), data) pdata, pflags = runner(meta, flagger, data) + if isinstance(pflags.columns, pd.MultiIndex): cols = (pflags .columns.get_level_values(0) .drop_duplicates()) assert (cols == [var1, var2]).all() + assert flagger.isFlagged(pflags[var2]).any() else: assert (pflags.columns == [var1, var2]).all() + assert flagger.isFlagged(pflags[var2]).any() @pytest.mark.parametrize("flagger", TESTFLAGGERS)