diff --git a/tests/flytekit/unit/core/test_type_conversion_errors.py b/tests/flytekit/unit/core/test_type_conversion_errors.py index dda19dd126..250e249611 100644 --- a/tests/flytekit/unit/core/test_type_conversion_errors.py +++ b/tests/flytekit/unit/core/test_type_conversion_errors.py @@ -1,10 +1,11 @@ """Unit tests for type conversion errors.""" +from datetime import timedelta from string import ascii_lowercase from typing import Tuple import pytest -from hypothesis import given +from hypothesis import given, settings from hypothesis import strategies as st from flytekit import task, workflow @@ -50,6 +51,7 @@ def wf_with_multioutput_error1(a: int, b: int) -> Tuple[str, int]: @given(st.booleans() | st.integers() | st.text(ascii_lowercase)) +@settings(deadline=timedelta(seconds=2)) def test_task_input_error(incorrect_input): with pytest.raises( TypeError, @@ -62,6 +64,7 @@ def test_task_input_error(incorrect_input): @given(st.floats()) +@settings(deadline=timedelta(seconds=2)) def test_task_output_error(correct_input): with pytest.raises( TypeError, @@ -74,6 +77,7 @@ def test_task_output_error(correct_input): @given(st.integers()) +@settings(deadline=timedelta(seconds=2)) def test_workflow_with_task_error(correct_input): with pytest.raises( TypeError, @@ -88,6 +92,7 @@ def test_workflow_with_task_error(correct_input): @given(st.booleans() | st.floats() | st.text(ascii_lowercase)) +@settings(deadline=timedelta(seconds=2)) def test_workflow_with_input_error(incorrect_input): with pytest.raises( TypeError, @@ -99,6 +104,7 @@ def test_workflow_with_input_error(incorrect_input): @given(st.integers()) +@settings(deadline=timedelta(seconds=2)) def test_workflow_with_output_error(correct_input): with pytest.raises( TypeError, @@ -118,6 +124,7 @@ def test_workflow_with_output_error(correct_input): ], ) @given(st.integers()) +@settings(deadline=timedelta(seconds=2)) def test_workflow_with_multioutput_error(workflow, position, correct_input): with pytest.raises( TypeError,