type hint fixes for adaptive/tests/test_learner1d.py

Author: basnijholt

adaptive/tests/test_learner1d.py

@@ -1,5 +1,5 @@
 import random
-from typing import Tuple, Union
+from typing import Tuple
 
 import numpy as np
 
@@ -105,9 +105,7 @@ def test_loss_interpolation() -> None:
         assert abs(expected_loss - loss) < 1e-15, (expected_loss, loss)
 
 
-def _run_on_discontinuity(
-    x_0: Union[int, float], bounds: Union[Tuple[int, int], Tuple[float, float]]
-) -> Learner1D:
+def _run_on_discontinuity(x_0: float, bounds: Tuple[float, float]) -> Learner1D:
     def f(x):
         return -1 if x < x_0 else +1
 
@@ -169,7 +167,7 @@ def goal(l):
     assert learner.npoints != 1000
 
 
-def small_deviations(x: float) -> Union[int, float]:
+def small_deviations(x: float) -> float:
     return 0 if x <= 1 else 1 + 10 ** (-random.randint(12, 14))