diff --git a/tests/test_deprecation_warnings.py b/tests/test_deprecation_warnings.py deleted file mode 100644 index dceb58c4..00000000 --- a/tests/test_deprecation_warnings.py +++ /dev/null @@ -1,25 +0,0 @@ -import warnings -from unittest.mock import patch -from src import ai_client -from src.result_types import Result - - -def test_send_deprecated_warning_emitted_once_per_site() -> None: - with patch.object(ai_client, "set_provider"): - with patch.object(ai_client, "_send_gemini", return_value=Result(data="x")): - with warnings.catch_warnings(record=True) as w: - warnings.simplefilter("always") - ai_client.send("s", "u") - ai_client.send("s", "u") - deprecation_warnings = [x for x in w if issubclass(x.category, DeprecationWarning)] - assert len(deprecation_warnings) >= 1 - - -def test_send_result_does_not_emit_deprecation() -> None: - with patch.object(ai_client, "set_provider"): - with patch.object(ai_client, "_send_gemini", return_value=Result(data="x")): - with warnings.catch_warnings(record=True) as w: - warnings.simplefilter("always") - ai_client.send_result("s", "u") - deprecation_warnings = [x for x in w if issubclass(x.category, DeprecationWarning)] - assert len(deprecation_warnings) == 0