Ajout type contrat

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

@@ -0,0 +1,56 @@
+import datetime
+
+import numpy as np
+import pytest
+
+from pandas._libs.tslibs import (
+    Resolution,
+    get_resolution,
+)
+from pandas._libs.tslibs.dtypes import NpyDatetimeUnit
+
+
+def test_get_resolution_nano():
+    # don't return the fallback RESO_DAY
+    arr = np.array([1], dtype=np.int64)
+    res = get_resolution(arr)
+    assert res == Resolution.RESO_NS
+
+
+def test_get_resolution_non_nano_data():
+    arr = np.array([1], dtype=np.int64)
+    res = get_resolution(arr, None, NpyDatetimeUnit.NPY_FR_us.value)
+    assert res == Resolution.RESO_US
+
+    res = get_resolution(arr, datetime.UTC, NpyDatetimeUnit.NPY_FR_us.value)
+    assert res == Resolution.RESO_US
+
+
+@pytest.mark.parametrize(
+    "freqstr,expected",
+    [
+        ("Y", "year"),
+        ("Q", "quarter"),
+        ("M", "month"),
+        ("D", "day"),
+        ("h", "hour"),
+        ("min", "minute"),
+        ("s", "second"),
+        ("ms", "millisecond"),
+        ("us", "microsecond"),
+        ("ns", "nanosecond"),
+    ],
+)
+def test_get_attrname_from_abbrev(freqstr, expected):
+    reso = Resolution.get_reso_from_freqstr(freqstr)
+    assert reso.attr_abbrev == freqstr
+    assert reso.attrname == expected
+
+
+@pytest.mark.parametrize("freq", ["H", "S"])
+def test_unit_H_S_raises(freq):
+    # GH#59143
+    msg = f"Invalid frequency: {freq}"
+
+    with pytest.raises(ValueError, match=msg):
+        Resolution.get_reso_from_freqstr(freq)