fix: init tests

tests/test_notebooks_run.py

@@ -1,31 +1,35 @@
 """
 collect all notebooks in examples, and check that they run without error
 """
+from __future__ import annotations
+
+from glob import glob
+from pathlib import Path
 
 import nbformat
+from matplotlib import pyplot as plt
 from nbconvert.preprocessors import ExecutePreprocessor
-from pathlib import Path
-from glob import glob
-from matplotlib import pyplot as plt  # this script doesn't use this but the notebooks do
+
 this_file_loc = Path(__file__).parent
 
-def check_notebook_runs(notebook_loc):
 
-    print(f'running: {notebook_loc}...')
+def check_notebook_runs(notebook_loc):
+    print(f"running: {notebook_loc}...")
     try:
-        with open(notebook_loc, encoding='utf8') as f:
+        with open(notebook_loc, encoding="utf8") as f:
             nb = nbformat.read(f, as_version=4)
-        ep = ExecutePreprocessor(timeout=600, kernel_name='python3')
-        ep.preprocess(nb, {'metadata': {'path': Path(notebook_loc).parent}})
+        ep = ExecutePreprocessor(timeout=600, kernel_name="python3")
+        ep.preprocess(nb, {"metadata": {"path": Path(notebook_loc).parent}})
     except Exception as e:
-        raise Exception(f'failed to run notebook {notebook_loc}')
-    print(f'success!')
+        raise Exception(f"failed to run notebook {notebook_loc}")
+    print(f"success!")
+
 
 def test_all_notebooks_run():
     # get all notebooks:
-    notebooks = glob(str(this_file_loc.parent / 'examples' / '*.ipynb'))
-    notebooks_not_to_run = ['put_notebooks_to_skip_here']
+    notebooks = glob(str(this_file_loc.parent / "examples" / "*.ipynb"))
+    notebooks_not_to_run = ["put_notebooks_to_skip_here"]
     for notebook in notebooks:
         if any([nb in notebook for nb in notebooks_not_to_run]):
             continue
-        check_notebook_runs(notebook)
+        check_notebook_runs(notebook)

tests/test_observer.py

@@ -1,13 +1,15 @@
-from bayes_opt.bayesian_optimization import Observable
-from bayes_opt.observer import _Tracker
-from bayes_opt.event import Events
+from __future__ import annotations
+
 import os
 
+from bayes_opt.bayesian_optimization import Observable
+from bayes_opt.event import Events
+from bayes_opt.observer import _Tracker
 
 EVENTS = ["a", "b", "c"]
 
 
-class SimpleObserver():
+class SimpleObserver:
     def __init__(self):
         self.counter = 0
 
@@ -20,13 +22,13 @@ def test_get_subscribers():
     observable = Observable(events=EVENTS)
     observable.subscribe("a", observer)
 
-    assert observer in observable.get_subscribers('a')
-    assert observer not in observable.get_subscribers('b')
-    assert observer not in observable.get_subscribers('c')
+    assert observer in observable.get_subscribers("a")
+    assert observer not in observable.get_subscribers("b")
+    assert observer not in observable.get_subscribers("c")
 
-    assert len(observable.get_subscribers('a')) == 1
-    assert len(observable.get_subscribers('b')) == 0
-    assert len(observable.get_subscribers('c')) == 0
+    assert len(observable.get_subscribers("a")) == 1
+    assert len(observable.get_subscribers("b")) == 0
+    assert len(observable.get_subscribers("c")) == 0
 
 
 def test_unsubscribe():
@@ -36,8 +38,8 @@ def test_unsubscribe():
     observable.subscribe("a", observer)
     observable.unsubscribe("a", observer)
 
-    assert observer not in observable.get_subscribers('a')
-    assert len(observable.get_subscribers('a')) == 0
+    assert observer not in observable.get_subscribers("a")
+    assert len(observable.get_subscribers("a")) == 0
 
 
 def test_dispatch():
@@ -51,17 +53,17 @@ def test_dispatch():
     assert observer_a.counter == 0
     assert observer_b.counter == 0
 
-    observable.dispatch('b')
+    observable.dispatch("b")
     assert observer_a.counter == 0
     assert observer_b.counter == 1
 
-    observable.dispatch('a')
-    observable.dispatch('b')
+    observable.dispatch("a")
+    observable.dispatch("b")
     assert observer_a.counter == 1
     assert observer_b.counter == 2
 
-    observable.dispatch('a')
-    observable.dispatch('c')
+    observable.dispatch("a")
+    observable.dispatch("c")
     assert observer_a.counter == 2
     assert observer_a.counter == 2
 
@@ -110,14 +112,15 @@ def max(self):
 
     tracker._time_metrics()
     assert start_time == tracker._start_time
-    if not 'win' in os.sys.platform:
+    if "win" not in os.sys.platform:
         assert previous_time < tracker._previous_time
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     r"""
     CommandLine:
         python tests/test_observer.py
     """
     import pytest
+
     pytest.main([__file__])

tests/test_queue.py

@@ -1,4 +1,7 @@
+from __future__ import annotations
+
 import pytest
+
 from bayes_opt.bayesian_optimization import Queue
 
 
@@ -19,7 +22,6 @@ def test_add():
 
 
 def test_queue():
-
     queue = Queue()
 
     with pytest.raises(StopIteration):
@@ -40,8 +42,7 @@ def test_queue():
     assert len(queue) == 0
 
 
-
-if __name__ == '__main__':
+if __name__ == "__main__":
     r"""
     CommandLine:
         python tests/test_observer.py