Merge pull request #36 from rcholic/minor_fix1

Make SessionHours optional
rcholic · Jul 20, 2024 · 4cd5409 · 4cd5409
2 parents beab2ba + 9c1e7a9
commit 4cd5409
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/cschwabpy/models/__init__.py b/cschwabpy/models/__init__.py
@@ -203,7 +203,7 @@ class Market(JSONSerializableBaseModel):
     product: str
     productName: str
     isOpen: bool
-    sessionHours: SessionHours
+    sessionHours: Optional[SessionHours] = None
 
 
 class EquityMarket(JSONSerializableBaseModel):

diff --git a/tests/test_models.py b/tests/test_models.py
@@ -97,6 +97,7 @@ async def test_market_hours(httpx_mock: HTTPXMock) -> None:
     all_market = MarketHourInfo(**all_market_json)
     assert all_market is not None
     assert all_market.equity is not None
+    assert all_market.equity.EQ.sessionHours is not None
     assert all_market.equity.EQ.sessionHours.regularMarket is not None
     assert all_market.equity.EQ.sessionHours.preMarket is not None