Update tests/lib/streaming/test_messages.py

dtmeadows · RobertCraigie · stainless-app[bot] · commit a5a32b84cf93 · 2025-06-02T07:27:04.000-04:00
Co-authored-by: Robert Craigie &lt;robert@craigie.dev&gt;
diff --git a/tests/lib/streaming/test_messages.py b/tests/lib/streaming/test_messages.py
@@ -389,10 +389,8 @@ def test_stream_method_definition_in_sync(sync: bool) -> None:
 
 # go through all the ContentBlock types to make sure the type alias is up to date
 # with any type that has an input property of type object
+@pytest.mark.skipif(not PYDANTIC_V2, reason='only applicable in pydantic v2')
 def test_tracks_tool_input_type_alias_is_up_to_date() -> None:
-    # only run this on Pydantic v2
-    if not PYDANTIC_V2:
-        pytest.skip("This test is only applicable for Pydantic v2")
     from typing import get_args
 
     from pydantic import BaseModel
