File size: 1,541 Bytes
460ec88 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
"""
Test runner for GAIA agent components.
This script runs tests for all GAIA agent components including:
- Answer formatter
- Integrated agent
- Multimodal processing
"""
import unittest
import os
import sys
# Ensure we can import from the parent directory
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
# Import test modules
from test_answer_formatter import TestAnswerFormatter
from test_integrated_agent import TestIntegratedAgent
if __name__ == "__main__":
print("Running GAIA agent tests...")
# Create the test loader
loader = unittest.TestLoader()
# Create a test suite including all tests
suite = unittest.TestSuite()
# Add the TestAnswerFormatter tests
print("Adding answer formatter tests...")
formatter_suite = loader.loadTestsFromTestCase(TestAnswerFormatter)
suite.addTest(formatter_suite)
# Add the TestIntegratedAgent tests
print("Adding integrated agent tests...")
integrated_suite = loader.loadTestsFromTestCase(TestIntegratedAgent)
suite.addTest(integrated_suite)
# Run the tests with verbosity
print("\nRunning all tests...\n")
runner = unittest.TextTestRunner(verbosity=2)
result = runner.run(suite)
# Report results
if result.wasSuccessful():
print("\nAll tests passed successfully!")
sys.exit(0)
else:
print("\nSome tests failed!")
print(f"Failures: {len(result.failures)}")
print(f"Errors: {len(result.errors)}")
sys.exit(1) |