charieswei
/
TradingAgents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update tests/test_y_finance_bulk_indicator.py 

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Sam Chen 2026-03-31 01:24:46 -05:00 committed by GitHub
parent 9524e04fe1
commit 474061fb5c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/test_y_finance_bulk_indicator.py
View File

@ -13,10 +13,10 @@ class YFinanceBulkIndicatorTests(unittest.TestCase):
fake_stockstats.wrap = lambda df: df fake_stockstats.wrap = lambda df: df
with patch.dict(sys.modules, {"stockstats": fake_stockstats}): with patch.dict(sys.modules, {"stockstats": fake_stockstats}):
sys.modules.pop("tradingagents.dataflows.stockstats_utils", None) with patch.dict(sys.modules):
sys.modules.pop("tradingagents.dataflows.y_finance", None) sys.modules.pop("tradingagents.dataflows.stockstats_utils", None)
sys.modules.pop("tradingagents.dataflows.y_finance", None)
try:
y_finance = importlib.import_module("tradingagents.dataflows.y_finance") y_finance = importlib.import_module("tradingagents.dataflows.y_finance")
sample_df = pd.DataFrame( sample_df = pd.DataFrame(
{ {
@ -27,12 +27,9 @@ class YFinanceBulkIndicatorTests(unittest.TestCase):
with patch.object(y_finance, "load_ohlcv", return_value=sample_df): with patch.object(y_finance, "load_ohlcv", return_value=sample_df):
result = y_finance._get_stock_stats_bulk("AAPL", "rsi", "2024-01-03") result = y_finance._get_stock_stats_bulk("AAPL", "rsi", "2024-01-03")
finally:
sys.modules.pop("tradingagents.dataflows.stockstats_utils", None)
sys.modules.pop("tradingagents.dataflows.y_finance", None)
self.assertEqual(result["2024-01-02"], "N/A") self.assertEqual(result["2024-01-02"], "N/A")
self.assertEqual(result["2024-01-03"], "55.5") self.assertEqual(result["2024-01-03"], "55.5")
if __name__ == "__main__": if __name__ == "__main__":