Updated script that can be controled by Nodejs web app

2024-11-25 12:24:18 +07:00
parent c440eda1f4
commit 8b0ab2bd3a
8662 changed files with 1803808 additions and 34 deletions
@@ -0,0 +1,27 @@
+import pytest
+
+from pandas._libs.tslibs.parsing import get_rule_month
+
+from pandas.tseries import offsets
+
+
+@pytest.mark.parametrize(
+    "obj,expected",
+    [
+        ("W", "DEC"),
+        (offsets.Week().freqstr, "DEC"),
+        ("D", "DEC"),
+        (offsets.Day().freqstr, "DEC"),
+        ("Q", "DEC"),
+        (offsets.QuarterEnd(startingMonth=12).freqstr, "DEC"),
+        ("Q-JAN", "JAN"),
+        (offsets.QuarterEnd(startingMonth=1).freqstr, "JAN"),
+        ("Y-DEC", "DEC"),
+        (offsets.YearEnd().freqstr, "DEC"),
+        ("Y-MAY", "MAY"),
+        (offsets.YearEnd(month=5).freqstr, "MAY"),
+    ],
+)
+def test_get_rule_month(obj, expected):
+    result = get_rule_month(obj)
+    assert result == expected