All permutations

PMBio · May 2, 2024 · 2c4c9cb · 2c4c9cb
1 parent 5435c95
commit 2c4c9cb
Showing 1 changed file with 57 additions and 0 deletions.
diff --git a/tests/deeprvat/test_associate.py b/tests/deeprvat/test_associate.py
@@ -52,6 +52,36 @@ def expected_array(request, tmp_path) -> Path:
             tests_data_dir / f"combine_burden_chunks/expected/burdens_{n_chunks}.zip",
         )
         for n_chunks in range(2, 5)
+    ]
+    + [
+        (
+            n_chunks,
+            True,
+            True,
+            tests_data_dir / f"combine_burden_chunks/input/chunks_{n_chunks}.zip",
+            tests_data_dir / f"combine_burden_chunks/expected/burdens_{n_chunks}.zip",
+        )
+        for n_chunks in range(2, 5)
+    ]
+    + [
+        (
+            n_chunks,
+            True,
+            False,
+            tests_data_dir / f"combine_burden_chunks/input/chunks_{n_chunks}.zip",
+            tests_data_dir / f"combine_burden_chunks/expected/burdens_{n_chunks}.zip",
+        )
+        for n_chunks in range(2, 5)
+    ]
+    + [
+        (
+            n_chunks,
+            False,
+            True,
+            tests_data_dir / f"combine_burden_chunks/input/chunks_{n_chunks}.zip",
+            tests_data_dir / f"combine_burden_chunks/expected/burdens_{n_chunks}.zip",
+        )
+        for n_chunks in range(2, 5)
     ],
     indirect=["chunks_data", "expected_array"],
 )
@@ -96,6 +126,33 @@ def test_combine_burden_chunks_data_same(
             tests_data_dir / f"combine_burden_chunks/input/chunks_{n_chunks}.zip",
         )
         for n_chunks in range(2, 5)
+    ]
+    + [
+        (
+            n_chunks,
+            True,
+            True,
+            tests_data_dir / f"combine_burden_chunks/input/chunks_{n_chunks}.zip",
+        )
+        for n_chunks in range(2, 5)
+    ]
+    + [
+        (
+            n_chunks,
+            True,
+            False,
+            tests_data_dir / f"combine_burden_chunks/input/chunks_{n_chunks}.zip",
+        )
+        for n_chunks in range(2, 5)
+    ]
+    + [
+        (
+            n_chunks,
+            False,
+            True,
+            tests_data_dir / f"combine_burden_chunks/input/chunks_{n_chunks}.zip",
+        )
+        for n_chunks in range(2, 5)
     ],
     indirect=["chunks_data"],
 )