Match all facets with keep=False

Author: bouweandela

packages/ref-core/src/cmip_ref_core/metrics.py

@@ -328,6 +328,7 @@ def apply_filters(self, data_catalog: pd.DataFrame) -> pd.DataFrame:
             Filtered data catalog
         """
         for facet_filter in self.filters:
+            mask = None
             for facet, value in facet_filter.facets.items():
                 clean_value = value if isinstance(value, tuple) else (value,)
 
@@ -336,7 +337,13 @@ def apply_filters(self, data_catalog: pd.DataFrame) -> pd.DataFrame:
                         f"Facet {facet!r} not in data catalog columns: {data_catalog.columns.to_list()}"
                     )
 
-                mask = data_catalog[facet].isin(clean_value)
+                facet_mask = data_catalog[facet].isin(clean_value)
+                if mask is None:
+                    mask = facet_mask
+                else:
+                    mask &= facet_mask
+
+            if mask is not None:
                 if not facet_filter.keep:
                     mask = ~mask
 

packages/ref-core/tests/unit/test_metrics.py

@@ -336,6 +336,40 @@ def test_apply_filters_dont_keep(apply_data_catalog):
     )
 
 
+def test_apply_filters_dont_keep_multifacet(apply_data_catalog):
+    """Test that all facet values must match to exclude a file from the catalog."""
+    requirement = DataRequirement(
+        source_type=SourceDatasetType.CMIP6,
+        filters=(
+            FacetFilter(
+                {
+                    "variable": "tas",
+                    "source_id": "CAS",
+                },
+                keep=False,
+            ),
+        ),
+        group_by=None,
+    )
+
+    filtered = requirement.apply_filters(apply_data_catalog)
+    pd.testing.assert_frame_equal(
+        filtered,
+        pd.DataFrame(
+            {
+                "variable": ["tas", "pr", "rsut", "tas"],
+                "source_id": [
+                    "CESM2",
+                    "CESM2",
+                    "CESM2",
+                    "ACCESS",
+                ],
+            },
+            index=[0, 1, 2, 3],
+        ),
+    )
+
+
 def test_apply_filters_missing(apply_data_catalog):
     requirement = DataRequirement(
         source_type=SourceDatasetType.CMIP6,