Ajout type contrat

venv/lib/python3.12/site-packages/pandas/tests/test_errors.py

@@ -0,0 +1,144 @@
+import warnings
+
+import pytest
+
+from pandas.errors import (
+    AbstractMethodError,
+    Pandas4Warning,
+    Pandas5Warning,
+    PandasChangeWarning,
+    PandasDeprecationWarning,
+    PandasPendingDeprecationWarning,
+    UndefinedVariableError,
+)
+
+import pandas as pd
+import pandas._testing as tm
+
+
+@pytest.mark.parametrize(
+    "exc",
+    [
+        "AttributeConflictWarning",
+        "CSSWarning",
+        "CategoricalConversionWarning",
+        "ClosedFileError",
+        "DataError",
+        "DatabaseError",
+        "DtypeWarning",
+        "EmptyDataError",
+        "IncompatibilityWarning",
+        "IndexingError",
+        "InvalidColumnName",
+        "InvalidComparison",
+        "InvalidVersion",
+        "LossySetitemError",
+        "MergeError",
+        "NoBufferPresent",
+        "NumExprClobberingError",
+        "NumbaUtilError",
+        "OptionError",
+        "OutOfBoundsDatetime",
+        "ParserError",
+        "ParserWarning",
+        "PerformanceWarning",
+        "PossibleDataLossError",
+        "PossiblePrecisionLoss",
+        "PyperclipException",
+        "SpecificationError",
+        "UnsortedIndexError",
+        "UnsupportedFunctionCall",
+        "ValueLabelTypeMismatch",
+    ],
+)
+def test_exception_importable(exc):
+    from pandas import errors
+
+    err = getattr(errors, exc)
+    assert err is not None
+
+    # check that we can raise on them
+
+    msg = "^$"
+
+    with pytest.raises(err, match=msg):
+        raise err()
+
+
+def test_catch_oob():
+    from pandas import errors
+
+    msg = "Cannot cast 1500-01-01 00:00:00 to unit='ns' without overflow"
+    with pytest.raises(errors.OutOfBoundsDatetime, match=msg):
+        pd.Timestamp("15000101").as_unit("ns")
+
+
+@pytest.mark.parametrize("is_local", [True, False])
+def test_catch_undefined_variable_error(is_local):
+    variable_name = "x"
+    if is_local:
+        msg = f"local variable '{variable_name}' is not defined"
+    else:
+        msg = f"name '{variable_name}' is not defined"
+
+    with pytest.raises(UndefinedVariableError, match=msg):
+        raise UndefinedVariableError(variable_name, is_local)
+
+
+class Foo:
+    @classmethod
+    def classmethod(cls):
+        raise AbstractMethodError(cls, methodtype="classmethod")
+
+    @property
+    def property(self):
+        raise AbstractMethodError(self, methodtype="property")
+
+    def method(self):
+        raise AbstractMethodError(self)
+
+
+def test_AbstractMethodError_classmethod():
+    xpr = "This classmethod must be defined in the concrete class Foo"
+    with pytest.raises(AbstractMethodError, match=xpr):
+        Foo.classmethod()
+
+    xpr = "This property must be defined in the concrete class Foo"
+    with pytest.raises(AbstractMethodError, match=xpr):
+        Foo().property
+
+    xpr = "This method must be defined in the concrete class Foo"
+    with pytest.raises(AbstractMethodError, match=xpr):
+        Foo().method()
+
+
+@pytest.mark.parametrize(
+    "warn_category, catch_category",
+    [
+        (Pandas4Warning, PandasChangeWarning),
+        (Pandas4Warning, PandasDeprecationWarning),
+        (Pandas5Warning, PandasChangeWarning),
+        (Pandas5Warning, PandasPendingDeprecationWarning),
+    ],
+)
+def test_pandas_warnings(warn_category, catch_category):
+    # https://github.com/pandas-dev/pandas/pull/61468
+    with tm.assert_produces_warning(catch_category):
+        warnings.warn("test", category=warn_category)
+
+
+@pytest.mark.parametrize(
+    "warn_category, filter_category",
+    [
+        (Pandas4Warning, PandasChangeWarning),
+        (Pandas4Warning, PandasDeprecationWarning),
+        (Pandas5Warning, PandasChangeWarning),
+        (Pandas5Warning, PandasPendingDeprecationWarning),
+    ],
+)
+def test_pandas_warnings_filter(warn_category, filter_category):
+    # https://github.com/pandas-dev/pandas/pull/61468
+    # Ensure users can suppress warnings.
+    with tm.assert_produces_warning(None), warnings.catch_warnings():
+        warnings.filterwarnings(category=filter_category, action="ignore")
+        warnings.warn("test", category=warn_category)