type hint fixes for adaptive/tests/algorithm_4.py

Author: basnijholt

adaptive/tests/algorithm_4.py

@@ -3,7 +3,6 @@
 
 from collections import defaultdict
 from fractions import Fraction
-from functools import partial
 from typing import Callable, List, Tuple, Union
 
 import numpy as np
@@ -247,7 +246,7 @@ def points(self) -> np.ndarray:
 
     @classmethod
     def make_first(
-        cls, f: Union[partial, Callable], a: int, b: int, depth: int = 2
+        cls, f: Callable, a: int, b: int, depth: int = 2
     ) -> Tuple["_Interval", int]:
         ival = _Interval(a, b, depth, 1)
         fx = f(ival.points())
@@ -269,7 +268,7 @@ def calc_igral_and_err(self, c_old: np.ndarray) -> float:
         return c_diff
 
     def split(
-        self, f: Union[partial, Callable]
+        self, f: Callable
     ) -> Union[Tuple[Tuple[float, float, float], int], Tuple[List["_Interval"], int]]:
         m = (self.a + self.b) / 2
         f_center = self.fx[(len(self.fx) - 1) // 2]
@@ -295,7 +294,7 @@ def split(
 
         return ivals, nr_points
 
-    def refine(self, f: Union[partial, Callable]) -> Tuple[np.ndarray, bool, int]:
+    def refine(self, f: Callable) -> Tuple[np.ndarray, bool, int]:
         """Increase degree of interval."""
         self.depth = depth = self.depth + 1
         points = self.points()
@@ -308,7 +307,7 @@ def refine(self, f: Union[partial, Callable]) -> Tuple[np.ndarray, bool, int]:
 
 
 def algorithm_4(
-    f: Union[partial, Callable], a: int, b: int, tol: float, N_loops: int = int(1e9)
+    f: Callable, a: int, b: int, tol: float, N_loops: int = int(1e9)
 ) -> Tuple[float, float, int, List["_Interval"]]:
     """ALGORITHM_4 evaluates an integral using adaptive quadrature. The
     algorithm uses Clenshaw-Curtis quadrature rules of increasing