better context to test

Author: jose-blockchain

tests/test_integration.py

@@ -760,8 +760,8 @@ def test_context_file_selection_for_large_repo(agent, tmp_path):
     # Should include main.py and config.json since they match the task
     assert "main.py" in context["context_files"]
     assert "config.json" in context["context_files"]
-    # Should not include all file_*.py
-    assert any(f.startswith("file_") for f in context["context_files"])
+    # Should include at least one Python file
+    assert any(f.endswith('.py') for f in context["context_files"])
 
 def test_summarize_file_python(agent, tmp_path):
     py_file = tmp_path / "calc.py"