pandas-market-calendars 4.6.0__tar.gz → 4.6.1__tar.gz

{pandas_market_calendars-4.6.0/pandas_market_calendars.egg-info → pandas_market_calendars-4.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pandas_market_calendars
-Version: 4.6.0
+Version: 4.6.1
 Summary: Market and exchange trading calendars for pandas
 Author-email: Ryan Sheftel <rsheftel@alumni.upenn.edu>
 License: MIT
@@ -19,7 +19,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Python: >=3.8
+Requires-Python: >=3.9
 Description-Content-Type: text/x-rst
 License-File: LICENSE
 License-File: NOTICE

{pandas_market_calendars-4.6.0 → pandas_market_calendars-4.6.1}/pandas_market_calendars/calendars/nyse.py

@@ -15,15 +15,14 @@
 
 from datetime import time
 from itertools import chain
+from typing import Literal, Union
 
 import pandas as pd
 from pandas.tseries.holiday import AbstractHolidayCalendar
 from pandas.tseries.offsets import CustomBusinessDay
 from pytz import timezone
 
-from typing import Literal, Union
 from pandas_market_calendars import calendar_utils as u
-
 from pandas_market_calendars.holidays.nyse import (
     # Always Celebrated Holidays
     USNewYearsDayNYSEpost1952,
@@ -1306,8 +1305,10 @@ class NYSEExchangeCalendar(MarketCalendar):
         :param tz: time zone in either string or pytz.timezone
         :return: DatetimeIndex of valid business days
         """
-        start_date = pd.Timestamp(start_date, tz=tz)
-        end_date = pd.Timestamp(end_date, tz=tz)
+        start_date = pd.Timestamp(start_date)
+        end_date = pd.Timestamp(end_date)
+        start_date = start_date.tz_convert(tz) if start_date.tz else start_date.tz_localize(tz)
+        end_date = end_date.tz_convert(tz) if end_date.tz else end_date.tz_localize(tz)
 
         # Starting Monday Sept. 29, 1952, no more saturday trading days
         if tz is None:

{pandas_market_calendars-4.6.0 → pandas_market_calendars-4.6.1/pandas_market_calendars.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pandas_market_calendars
-Version: 4.6.0
+Version: 4.6.1
 Summary: Market and exchange trading calendars for pandas
 Author-email: Ryan Sheftel <rsheftel@alumni.upenn.edu>
 License: MIT
@@ -19,7 +19,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Python: >=3.8
+Requires-Python: >=3.9
 Description-Content-Type: text/x-rst
 License-File: LICENSE
 License-File: NOTICE

{pandas_market_calendars-4.6.0 → pandas_market_calendars-4.6.1}/pyproject.toml

@@ -1,12 +1,12 @@
 [project]
 name = "pandas_market_calendars"
-version = "4.6.0"
+version = "4.6.1"
 authors = [
     { name = "Ryan Sheftel", email = "rsheftel@alumni.upenn.edu" },
 ]
 description = "Market and exchange trading calendars for pandas"
 readme = "README.rst"
-requires-python = ">=3.8"
+requires-python = ">=3.9"
 keywords = ["trading", "exchanges", "markets", "OTC", "datetime", "holiday", "business days"]
 license = { text = "MIT" }
 classifiers = [

{pandas_market_calendars-4.6.0 → pandas_market_calendars-4.6.1}/tests/test_nyse_calendar.py

@@ -1,5 +1,6 @@
 import datetime as dt
 import os
+from zoneinfo import ZoneInfo
 
 import pandas as pd
 import pytest
@@ -146,6 +147,20 @@ def test_valid_days():
     for tz in ("America/New_York", "Europe/Berlin", None):
         assert (valid.tz_localize(tz) == cal.valid_days(start, end, tz=tz)).all()
 
+    # test with dates with timezones attached
+    start = pd.Timestamp("2000-01-01", tz="America/New_York")
+    end = pd.Timestamp("2000-01-30", tz="America/New_York")
+    valid_w_tz = cal.valid_days(start, end, tz="UTC").tz_localize(None)
+    assert_index_equal(valid, valid_w_tz)
+
+
+def test_valid_days_tz_aware():
+    calendar = NYSEExchangeCalendar()
+    data_date = dt.datetime.strptime("20250121", "%Y%m%d").astimezone(ZoneInfo("UTC"))
+    actual = calendar.valid_days(data_date, data_date + dt.timedelta(days=7), tz="UTC")
+    expected = pd.bdate_range("2025-01-21", periods=6, tz="UTC")
+    assert_index_equal(actual, expected)
+
 
 def test_time_zone():
     assert NYSEExchangeCalendar().tz == pytz.timezone("America/New_York")