Add type-hints to adaptive/tests/test_average_learner1d.py

Author: basnijholt

adaptive/tests/test_average_learner1d.py

@@ -1,6 +1,6 @@
-from itertools import chain
-
 import numpy as np
+import pandas as pd
+from pandas.testing import assert_series_equal
 
 from adaptive import AverageLearner1D
 from adaptive.tests.test_learners import (
@@ -11,27 +11,13 @@
 
 
 def almost_equal_dicts(a, b):
-    assert a.keys() == b.keys()
-    for k, v1 in a.items():
-        v2 = b[k]
-        if (
-            v1 is None
-            or v2 is None
-            or isinstance(v1, (tuple, list))
-            and any(x is None for x in chain(v1, v2))
-        ):
-            assert v1 == v2
-        else:
-            try:
-                np.testing.assert_almost_equal(v1, v2)
-            except TypeError:
-                raise AssertionError(f"{v1} != {v2}")
+    assert_series_equal(pd.Series(sorted(a.items())), pd.Series(sorted(b.items())))
 
 
 def test_tell_many_at_point():
     f = generate_random_parametrization(noisy_peak)
     learner = AverageLearner1D(f, bounds=(-2, 2))
-    control = learner.new()
+    control = AverageLearner1D(f, bounds=(-2, 2))
     learner._recompute_losses_factor = 1
     control._recompute_losses_factor = 1
     simple_run(learner, 100)