Added hasattr assertions to tests/unit_tests/chains/test_graph_qa.py

langchain-ai · Dec 8, 2024 · ebb8b46 · ebb8b46
1 parent 573bbf4
commit ebb8b46
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/libs/neo4j/tests/unit_tests/chains/test_graph_qa.py b/libs/neo4j/tests/unit_tests/chains/test_graph_qa.py
@@ -76,8 +76,15 @@ def test_graph_cypher_qa_chain_prompt_selection_1() -> None:
         cypher_prompt=cypher_prompt,
         allow_dangerous_requests=True,
     )
+<<<<<<< HEAD
     assert chain.qa_chain.prompt == qa_prompt  # type: ignore[union-attr]
     assert chain.cypher_generation_chain.prompt == cypher_prompt
+=======
+    assert hasattr(chain.qa_chain, "first")
+    assert chain.qa_chain.first == qa_prompt
+    assert hasattr(chain.cypher_generation_chain, "first")
+    assert chain.cypher_generation_chain.first == cypher_prompt
+>>>>>>> 63f27b1 (Added hasattr assertions to tests/unit_tests/chains/test_graph_qa.py)
 
 
 def test_graph_cypher_qa_chain_prompt_selection_2() -> None: