26 lines
1.2 KiB
Python
26 lines
1.2 KiB
Python
import unittest
|
|
from pathlib import Path
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
class PromptToolConsistencyTests(unittest.TestCase):
|
|
def test_social_prompt_matches_real_tool_signatures(self):
|
|
source = (ROOT / "tradingagents" / "agents" / "analysts" / "social_media_analyst.py").read_text(encoding="utf-8")
|
|
self.assertIn("get_social_sentiment(symbol, start_date, end_date)", source)
|
|
self.assertIn("get_company_news(symbol, start_date, end_date)", source)
|
|
self.assertNotIn("get_news(query, start_date, end_date)", source)
|
|
self.assertNotIn("social media posts", source.lower())
|
|
|
|
def test_news_prompt_matches_real_tool_signatures(self):
|
|
source = (ROOT / "tradingagents" / "agents" / "analysts" / "news_analyst.py").read_text(encoding="utf-8")
|
|
self.assertIn("get_company_news(symbol, start_date, end_date)", source)
|
|
self.assertIn("get_macro_news(curr_date, look_back_days, limit, region, language)", source)
|
|
self.assertIn("get_disclosures(symbol, start_date, end_date)", source)
|
|
self.assertNotIn("get_news(query, start_date, end_date)", source)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|