test_rag.py formatting