!squash typings

Author: tony

tests/test_test.py

@@ -2,14 +2,15 @@
 
 import pytest
 
-from libtmux.test import WaitTimeout, retry_until
+from libtmux.exc import WaitTimeout
+from libtmux.test import retry_until
 
 
-def test_retry_three_times():
+def test_retry_three_times() -> None:
     ini = time()
     value = 0
 
-    def call_me_three_times():
+    def call_me_three_times() -> bool:
         nonlocal value
 
         if value == 2:
@@ -26,10 +27,10 @@ def call_me_three_times():
     assert abs((end - ini) - 0.1) < 0.01
 
 
-def test_function_times_out():
+def test_function_times_out() -> None:
     ini = time()
 
-    def never_true():
+    def never_true() -> bool:
         return False
 
     with pytest.raises(WaitTimeout):
@@ -40,10 +41,10 @@ def never_true():
     assert abs((end - ini) - 1.0) < 0.01
 
 
-def test_function_times_out_no_rise():
+def test_function_times_out_no_rise() -> None:
     ini = time()
 
-    def never_true():
+    def never_true() -> bool:
         return False
 
     retry_until(never_true, 1, raises=False)
@@ -53,10 +54,10 @@ def never_true():
     assert abs((end - ini) - 1.0) < 0.01
 
 
-def test_function_times_out_no_raise_assert():
+def test_function_times_out_no_raise_assert() -> None:
     ini = time()
 
-    def never_true():
+    def never_true() -> bool:
         return False
 
     assert not retry_until(never_true, 1, raises=False)
@@ -66,11 +67,11 @@ def never_true():
     assert abs((end - ini) - 1.0) < 0.01
 
 
-def test_retry_three_times_no_raise_assert():
+def test_retry_three_times_no_raise_assert() -> None:
     ini = time()
     value = 0
 
-    def call_me_three_times():
+    def call_me_three_times() -> bool:
         nonlocal value
 
         if value == 2: