Make deprecation warning catch more general

PrefectHQ · Feb 2, 2024 · 92ea4cf · 92ea4cf
1 parent fd1f553
commit 92ea4cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/results/test_flow_results.py b/tests/results/test_flow_results.py
@@ -435,7 +435,7 @@ def test_flow_server_state_schema_result_is_respected(persist_result, return_sta
     def my_flow():
         return return_state
 
-    with pytest.warns(DeprecationWarning, match="Use `prefect.states.State` instead"):
+    with pytest.warns(DeprecationWarning, match="`prefect.states.State`"):
         state = my_flow(return_state=True)
 
     assert state.type == return_state.type
@@ -449,7 +449,7 @@ def my_flow():
     ) == return_state.dict(exclude={"id", "timestamp", "state_details", "data"})
 
     if return_state.data:
-        with pytest.warns(DeprecationWarning, match="use `prefect.states.State`"):
+        with pytest.warns(DeprecationWarning, match="`prefect.states.State`"):
             assert state.result(raise_on_failure=False) == return_state.data