minor changes to increase coverage

dstl · Apr 8, 2024 · 1521e2f · 1521e2f
1 parent 8e4fd44
commit 1521e2f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/stonesoup/tracker/base.py b/stonesoup/tracker/base.py
@@ -51,7 +51,7 @@ class _TrackerMixInNext(_TrackerMixInBase):
 
     @abstractmethod
     def __next__(self) -> Tuple[datetime.datetime, Set[Track]]:
-        ...
+        """Pull detections from the detector (`detector_iter`). Act on them to create tracks."""
 
     def update_tracker(self, time: datetime.datetime, detections: Set[Detection]) \
             -> Tuple[datetime.datetime, Set[Track]]:
@@ -73,4 +73,4 @@ def __next__(self) -> Tuple[datetime.datetime, Set[Track]]:
     @abstractmethod
     def update_tracker(self, time: datetime.datetime, detections: Set[Detection]) \
             -> Tuple[datetime.datetime, Set[Track]]:
-        ...
+        """Use `time` and `detections` to create tracks."""
diff --git a/stonesoup/tracker/tests/test_base.py b/stonesoup/tracker/tests/test_base.py
@@ -88,6 +88,13 @@ def test_tracker_without_detector_iter_error(tracker_class):
         next(tracker_without_detector)
 
 
+@pytest.mark.parametrize("tracker_class", [TrackerNextWithDetector, TrackerUpdateWithDetector])
+def test_tracker_without_detector_iter_error(tracker_class):
+    tracker = tracker_class(detector=None)
+    with pytest.raises(AttributeError):
+        iter(tracker)
+
+
 @pytest.mark.parametrize("tracker_class", [TrackerNextWithDetector, TrackerUpdateWithDetector])
 def test_tracker_with_detector_iter(tracker_class):
     tracker = tracker_class()