Add typing test

hynek · Aug 18, 2024 · 214165a · 214165a
1 parent daa62e5
commit 214165a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/typing/api.py b/tests/typing/api.py
@@ -17,9 +17,11 @@
     BoundRetryingCaller,
     RetryingCaller,
     is_active,
+    is_testing,
     retry,
     retry_context,
     set_active,
+    set_testing,
 )
 from stamina.instrumentation import (
     RetryDetails,
@@ -94,6 +96,14 @@ def exc_tune_waiting_timedelta() -> None: ...
 
 set_active(False)
 
+set_testing(True)
+set_testing(True, attempts=2)
+
+if is_testing() is False:
+    ...
+
+set_testing(False)
+
 
 def hook(details: RetryDetails) -> None:
     return None