Expand test case partitioning columns

ratt-ru · Jan 24, 2025 · dfce2a0 · dfce2a0
1 parent eac9cbe
commit dfce2a0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_backend.py b/tests/test_backend.py
@@ -118,7 +118,9 @@ def test_open_dataset(simmed_ms):
   ],
   ids=id_string,
 )
-@pytest.mark.parametrize("partition_columns", [["DATA_DESC_ID", "FIELD_ID"]])
+@pytest.mark.parametrize(
+  "partition_columns", [["DATA_DESC_ID", "OBSERVATION_ID", "FIELD_ID"]]
+)
 def test_open_dataset_partition_keys(
   simmed_ms, partition_columns, partition_key, pols, nfreq
 ):