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