Skip to content

Commit ff97779

Browse files
committed
Fix sleep_fn return type for mypy
1 parent bdaacd8 commit ff97779

File tree

2 files changed

+12
-6
lines changed

2 files changed

+12
-6
lines changed

tests/test_query.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,9 +67,12 @@ def test_default_timeout(
6767
freeze_time() as frozen_datetime,
6868
MockVWS(
6969
response_delay_seconds=response_delay_seconds,
70-
sleep_fn=lambda seconds: frozen_datetime.tick(
71-
delta=datetime.timedelta(seconds=seconds),
72-
),
70+
sleep_fn=lambda seconds: (
71+
frozen_datetime.tick(
72+
delta=datetime.timedelta(seconds=seconds),
73+
),
74+
None,
75+
)[1],
7376
) as mock,
7477
):
7578
database = VuforiaDatabase()

tests/test_vws.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -114,9 +114,12 @@ def test_default_timeout(
114114
freeze_time() as frozen_datetime,
115115
MockVWS(
116116
response_delay_seconds=response_delay_seconds,
117-
sleep_fn=lambda seconds: frozen_datetime.tick(
118-
delta=datetime.timedelta(seconds=seconds),
119-
),
117+
sleep_fn=lambda seconds: (
118+
frozen_datetime.tick(
119+
delta=datetime.timedelta(seconds=seconds),
120+
),
121+
None,
122+
)[1],
120123
) as mock,
121124
):
122125
database = VuforiaDatabase()

0 commit comments

Comments
 (0)