Merge pull request #325 from python-adaptive/learner1d-type-hints

Learner1D type hints and add typeguard to pytest tests

Author: basnijholt

adaptive/learner/average_learner1D.py

@@ -3,7 +3,17 @@
 from collections import defaultdict
 from copy import deepcopy
 from math import hypot
-from typing import Callable, DefaultDict, Dict, List, Optional, Sequence, Set, Tuple
+from typing import (
+    Callable,
+    DefaultDict,
+    Dict,
+    Iterable,
+    List,
+    Optional,
+    Sequence,
+    Set,
+    Tuple,
+)
 
 import numpy as np
 import scipy.stats
@@ -356,7 +366,7 @@ def _update_losses_resampling(self, x: Real, real=True) -> None:
         if (b is not None) and right_loss_is_unknown:
             self.losses_combined[x, b] = float("inf")
 
-    def _calc_error_in_mean(self, ys: Sequence[Real], y_avg: Real, n: int) -> float:
+    def _calc_error_in_mean(self, ys: Iterable[Real], y_avg: Real, n: int) -> float:
         variance_in_mean = sum((y - y_avg) ** 2 for y in ys) / (n - 1)
         t_student = scipy.stats.t.ppf(1 - self.alpha, df=n - 1)
         return t_student * (variance_in_mean / n) ** 0.5

adaptive/learner/base_learner.py

@@ -5,7 +5,7 @@
 from adaptive.utils import _RequireAttrsABCMeta, load, save
 
 
-def uses_nth_neighbors(n):
+def uses_nth_neighbors(n: int):
     """Decorator to specify how many neighboring intervals the loss function uses.
 
     Wraps loss functions to indicate that they expect intervals together