cr2: add plot_allfreqs()
diff --git a/tests/test_plot_utils.py b/tests/test_plot_utils.py
index 8deab38..e4403fa 100644
--- a/tests/test_plot_utils.py
+++ b/tests/test_plot_utils.py
@@ -2,6 +2,8 @@
import unittest
+from test_thermal import TestThermalBase
+import cr2
import plot_utils
class TestPlotUtils(unittest.TestCase):
@@ -25,3 +27,13 @@
plot_utils.post_plot_setup(ax)
plot_utils.post_plot_setup(ax, title="Foo")
plot_utils.post_plot_setup(ax, ylim=(0, 72))
+
+class TestPlotUtilsNeedTrace(TestThermalBase):
+ def test_plot_allfreqs(self):
+ """Test that plot_allfreqs() doesn't bomb"""
+
+ inp = cr2.InPower()
+ outp = cr2.OutPower()
+ map_label = {"0000000f": "A7", "000000f0": "A15"}
+
+ plot_utils.plot_allfreqs(inp, outp, map_label)