Skip to content

Commit

Permalink
Tests: Configure two flaky tests for re-running
Browse files Browse the repository at this point in the history
  • Loading branch information
amotl committed Dec 14, 2024
1 parent 1f4e0bc commit cca3956
Show file tree
Hide file tree
Showing 4 changed files with 20 additions and 1 deletion.
17 changes: 16 additions & 1 deletion poetry.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 1 addition & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,7 @@ pytest-socket = "^0.7.0"
pytest-watcher = "^0.3.4"
langchain-community = "^0.3.11"
langchain-tests = "==0.3.7"
pytest-rerunfailures = "^15.0"

[tool.poetry.group.codespell.dependencies]
codespell = "^2.2.6"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -184,6 +184,7 @@ def test_cratedb_with_filter_match(engine: sa.Engine) -> None:
assert scores == (1.0,)


@pytest.mark.flaky(reruns=5)
def test_cratedb_with_filter_distant_match(engine: sa.Engine) -> None:
"""Test end to end construction and search."""
texts = ["foo", "bar", "baz"]
Expand Down Expand Up @@ -310,6 +311,7 @@ def test_cratedb_collection_no_embedding_dimension(
)


@pytest.mark.flaky(reruns=5)
@pytest.mark.parametrize("operator", ["$in", "IN"])
def test_cratedb_with_filter_in_set(engine: sa.Engine, operator: str) -> None:
"""Test end to end construction and search."""
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
from tests.integration_tests.vectorstore.util import prune_document_ids


@pytest.mark.flaky(reruns=5)
def test_cratedb_multicollection_search_success(engine: sa.Engine) -> None:
"""
`CrateDBVectorStoreMultiCollection` provides functionality for
Expand Down

0 comments on commit cca3956

Please sign in to comment.