chore: Add debug printing to RAG tests

Author: chalmerlowe

tests/unit/vertex_rag/test_rag_data.py

@@ -419,6 +419,10 @@ def create_transformation_config(
 from vertexai.rag.utils.resources import RagVectorDbConfig
 
 def rag_corpus_eq(returned_corpus, expected_corpus):
+    if returned_corpus != expected_corpus:
+        print(f"Returned Corpus: {returned_corpus}")
+        print(f"Expected Corpus: {expected_corpus}")
+
     assert returned_corpus.name == expected_corpus.name
     assert returned_corpus.display_name == expected_corpus.display_name
 

tests/unit/vertex_rag/test_rag_data_preview.py

@@ -691,6 +691,10 @@ def create_transformation_config(
 from vertexai.rag.utils.resources import RagVectorDbConfig
 
 def rag_corpus_eq(returned_corpus, expected_corpus):
+    if returned_corpus != expected_corpus:
+        print(f"Returned Corpus: {returned_corpus}")
+        print(f"Expected Corpus: {expected_corpus}")
+
     assert returned_corpus.name == expected_corpus.name
     assert returned_corpus.display_name == expected_corpus.display_name
     assert returned_corpus.vector_db == expected_corpus.vector_db