diff --git a/requirements.txt b/requirements.txt index c8d3df7..b2fc8be 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,4 +4,6 @@ openai==1.41.0 python-dotenv==1.0.1 discord.py==2.4.0 anthropic==0.34.0 -GitPython==3.1.43 \ No newline at end of file +GitPython==3.1.43 +pytest +pytest-cov \ No newline at end of file diff --git a/tests/chatgpt/__init__.py b/tests/chatgpt/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/claude/__init__.py b/tests/claude/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/claude/test_base_telegram_inference_bot.py b/tests/claude/test_base_telegram_inference_bot.py new file mode 100644 index 0000000..8aa471f --- /dev/null +++ b/tests/claude/test_base_telegram_inference_bot.py @@ -0,0 +1,17 @@ +import unittest +from base_telegram_inference_bot import BaseTelegramInferenceBot + +class TestBaseTelegramInferenceBot(unittest.TestCase): + def setUp(self): + # Initialize the bot or mock any dependencies here + self.bot = BaseTelegramInferenceBot() + + def test_load_system_prompt(self): + # Example test case for load_system_prompt method + result = self.bot.load_system_prompt() + self.assertIsNotNone(result) # Replace with actual expected result + + # Additional test cases can be added here + +if __name__ == '__main__': + unittest.main() diff --git a/tests/integration/__init__.py b/tests/integration/__init__.py new file mode 100644 index 0000000..e69de29