replace flaky with pytest-rerunfailures (#7298)

deepset-ai · Mar 4, 2024 · 721691c · 721691c
1 parent f49dbdc
commit 721691c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -67,11 +67,11 @@ dependencies = [
  # Type check
  "mypy",
  # Test
- "pytest<8.1.0",
+ "pytest",
  "pytest-cov",
  "pytest-custom_exit_code", # used in the CI
  "pytest-asyncio",
- "flaky",
+ "pytest-rerunfailures",
  "responses",
  "tox",
  "coverage",

diff --git a/test/core/pipeline/test_draw.py b/test/core/pipeline/test_draw.py
@@ -1,10 +1,8 @@
 # SPDX-FileCopyrightText: 2022-present deepset GmbH <[email protected]>
 #
 # SPDX-License-Identifier: Apache-2.0
-import time
 from unittest.mock import MagicMock, patch
 
-import flaky
 import pytest
 import requests
 
@@ -14,7 +12,7 @@
 from haystack.testing.sample_components import AddFixedValue, Double
 
 
-@flaky.flaky(max_runs=5, rerun_filter=lambda *_: time.sleep(5))
+@pytest.mark.flaky(reruns=5, reruns_delay=5)
 @pytest.mark.integration
 def test_to_mermaid_image():
  pipe = Pipeline()