tests for concatDios

tests/lib/test_tools.py

@@ -3,6 +3,7 @@
 # SPDX-License-Identifier: GPL-3.0-or-later
 import pytest
 
+from dios import DictOfSeries as DoS
 import saqc.lib.tools as tools
 import pandas as pd
 import numpy as np
@@ -68,3 +69,33 @@ def test_toSequence(value, expected):
 def test_squeezeSequence(value, expected):
     result = tools.squeezeSequence(value)
     assert result == expected
+
+
+@pytest.mark.parametrize(
+    "data, expected",
+    [
+        # 2c + 1c -> 3c
+        ([DoS(dict(a=[1], b=[2])), DoS(dict(c=[3]))], DoS(dict(a=[1], b=[2], c=[3]))),
+        # 1c + 1c + 1c -> 3c
+        (
+            [DoS(dict(a=[1])), DoS(dict(b=[1])), DoS(dict(c=[1]))],
+            DoS(dict(a=[1], b=[1], c=[1])),
+        ),
+        # 2c + 1c (overwrite) = 2c
+        ([DoS(dict(a=[1], b=[2])), DoS(dict(b=[22]))], DoS(dict(a=[1], b=[22]))),
+        # 1c + 1c + 1c (all overwrite) -> 1c
+        (
+            [DoS(dict(a=[1])), DoS(dict(a=[11])), DoS(dict(a=[111]))],
+            DoS(dict(a=[111])),
+        ),
+    ],
+)
+def test_concatDios(data, expected):
+    result = tools.concatDios(data, warn=False)
+    assert result == expected
+
+
+@pytest.mark.parametrize("data", [[DoS(dict(a=[1], b=[2])), DoS(dict(b=[22]))]])
+def test_concatDios_warning(data):
+    with pytest.warns(UserWarning):
+        tools.concatDios(data, warn=True, stacklevel=0)