From 225a122aeee704dba4f686bcf03bac0151d10e97 Mon Sep 17 00:00:00 2001
From: David Schaefer <david.schaefer@ufz.de>
Date: Tue, 12 Mar 2024 10:53:48 +0100
Subject: [PATCH] fixed failing test

---
 saqc/__main__.py              |  2 +-
 tests/cli/test_integration.py | 17 ++++++++---------
 2 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/saqc/__main__.py b/saqc/__main__.py
index 9d1738b74..4aea86013 100644
--- a/saqc/__main__.py
+++ b/saqc/__main__.py
@@ -157,7 +157,7 @@ def main(
                 flagscol = flagscol.rename("flags")
             out[k] = pd.concat([data_result[k].rename("data"), flagscol], axis=1)
 
-        writeData(writer, out.to_pandas(), outfile)
+        writeData(writer, out.to_pandas(fill_value=-9999 if scheme == "positional" else np.nan), outfile)
 
 
 if __name__ == "__main__":
diff --git a/tests/cli/test_integration.py b/tests/cli/test_integration.py
index 0971c44b8..a2cf2a70c 100644
--- a/tests/cli/test_integration.py
+++ b/tests/cli/test_integration.py
@@ -38,13 +38,13 @@ POSITIONAL = [
     ",Battery,Battery,SM1,SM1,SM2,SM2\n",
     ",data,flags,data,flags,data,flags\n",
     "Date,,,,,,\n",
-    "2016-04-01 00:00:00,nan,-9999,nan,-9999,29.3157,90000\n",
-    "2016-04-01 00:05:48,3573.0,9,32.685,90,nan,-9999\n",
-    "2016-04-01 00:15:00,nan,-9999,nan,-9999,29.3157,90000\n",
-    "2016-04-01 00:20:42,3572.0,9,32.7428,90,nan,-9999\n",
-    "2016-04-01 00:30:00,nan,-9999,nan,-9999,29.3679,90002\n",
-    "2016-04-01 00:35:37,3572.0,9,32.6186,90,nan,-9999\n",
-    "2016-04-01 00:45:00,nan,-9999,nan,-9999,29.3679,90000\n",
+    "2016-04-01 00:00:00,-9999,-9999,-9999.0,-9999,29.3157,90000\n",
+    "2016-04-01 00:05:48,3573,9,32.685,90,-9999.0,-9999\n",
+    "2016-04-01 00:15:00,-9999,-9999,-9999.0,-9999,29.3157,90000\n",
+    "2016-04-01 00:20:42,3572,9,32.7428,90,-9999.0,-9999\n",
+    "2016-04-01 00:30:00,-9999,-9999,-9999.0,-9999,29.3679,90002\n",
+    "2016-04-01 00:35:37,3572,9,32.6186,90,-9999.0,-9999\n",
+    "2016-04-01 00:45:00,-9999,-9999,-9999.0,-9999,29.3679,90000\n",
 ]
 
 DMP = [
@@ -91,5 +91,4 @@ def test__main__py(tmp_path, scheme, expected):
     assert result.exit_code == 0, result.output
     with open(outfile, "r") as f:
         result = f.readlines()[:10]
-        print(result[4])
-        # assert result == expected
+        assert result == expected
-- 
GitLab