Create test_agent.py