Update agent_cache_test.py

added TODO