Add RAG Plotter Tests

Author: csherman-dstl

stonesoup/tests/test_plotter.py

@@ -1,5 +1,6 @@
 import warnings
 from datetime import datetime, timedelta
+from dataclasses import dataclass
 
 import matplotlib.pyplot as plt
 import numpy as np
@@ -14,7 +15,7 @@
 from stonesoup.platform.base import Obstacle
 from stonesoup.platform.shape import Shape
 from stonesoup.plotter import Plotter, Dimension, AnimatedPlotterly, AnimationPlotter, Plotterly, \
-    PolarPlotterly, AnimatedPolarPlotterly, merge_dicts
+    PolarPlotterly, AnimatedPolarPlotterly, RAGPlotterly, RAG, merge_dicts
 from stonesoup.predictor.kalman import KalmanPredictor
 from stonesoup.sensor.radar.radar import RadarElevationBearingRange
 from stonesoup.types.detection import TrueDetection, Clutter
@@ -686,3 +687,21 @@ def test_obstacles(plotters, obstacles):
         plotters.plot_measurements(all_measurements, [0, 1])
         plotters.plot_tracks(track, [0, 1])
         plotters.plot_obstacles(obstacles)
+
+
+@dataclass
+class TestMetric:
+    value: float
+    timestamp: int = None
+
+def test_rag_plot():
+    track = Track([State([0, 0], timestamp=0),
+                   State([1, 1], timestamp=1),
+                   State([2, 2], timestamp=2)], id="track_1")
+    plotter = RAGPlotterly(metric_name="Test Metric", target_value=0,
+                           rag_boundaries=RAG(GREEN=1, AMBER=2))
+    metrics = {("track_1", 1): {"Test Metric": TestMetric(value=[
+        TestMetric(value=0.5, timestamp=0),
+        TestMetric(value=1.5, timestamp=1),
+        TestMetric(value=2.5, timestamp=2)])}}
+    plotter.plot_tracks({track}, [0, 1], metrics)