From 52623febe1687c8631e221b2a6f43ef8713d429a Mon Sep 17 00:00:00 2001
From: Bert Palm <bert.palm@ufz.de>
Date: Tue, 16 Apr 2019 14:56:49 +0200
Subject: [PATCH] extended assign test
---
test/test_core.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/test/test_core.py b/test/test_core.py
index 20c592b6a..5f6b0c805 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)
--
GitLab