fix: pytest retry for flaky test (#25)

* fix: pytest retry for flaky test

* chore: minor updates (delay and retry on AssertError only)

---------

Co-authored-by: gRedHeadphone <[email protected]>

Author: RedHeadphone

pyproject.toml

@@ -46,6 +46,7 @@ test = [
     "isort==5.13.2",
     "mypy==1.16.0",
     "pytest==8.3.3",
+    "pytest-retry==1.7.0",
     "pytest-xdist==3.6.1",
     "pytest-cov==5.0.0",
     "pytest-asyncio==0.24.0",

tests/test_graph_retriever.py

@@ -203,6 +203,7 @@ def retrievers_dynamic():
     graph_store.clean_up()
 
 
+@pytest.mark.flaky(retries=3, only_on=[AssertionError], delay=1)
 def test_graph_retriever2_static(retrievers_static):
     """Test the graph retriever."""
     for retriever in retrievers_static:
@@ -213,6 +214,7 @@ def test_graph_retriever2_static(retrievers_static):
         assert "Elias Thorne" in str(res)
 
 
+@pytest.mark.flaky(retries=3, only_on=[AssertionError], delay=1)
 def test_graph_retriever2_dynamic(retrievers_dynamic):
     """Test the graph retriever."""
     retriever = retrievers_dynamic[0]  # only NL2GQL fails