Update warning catching tests

stac-utils · Feb 1, 2024 · 41a0446 · 41a0446
1 parent 3d8315d
commit 41a0446
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -405,8 +405,9 @@ def test_get_collections_without_conformance_fallsback_to_pystac(
         # Mock the collection
         requests_mock.get(pc_collection_href, status_code=200, json=pc_collection_dict)
 
-        with pytest.warns(DoesNotConformTo, match="COLLECTIONS, FEATURES"):
-            _ = next(api.get_collections())
+        with pytest.warns(FallbackToPystac):
+            with pytest.warns(DoesNotConformTo, match="COLLECTIONS, FEATURES"):
+                _ = next(api.get_collections())
 
         history = requests_mock.request_history
         assert len(history) == 2