Add basic test for plot.py
Change-Id: I976dffaf7ca04b4403e29d6762aebd94afe86739
diff --git a/frc971/analysis/BUILD b/frc971/analysis/BUILD
index 82aca97..2686bcd 100644
--- a/frc971/analysis/BUILD
+++ b/frc971/analysis/BUILD
@@ -69,3 +69,13 @@
"@matplotlib_repo//:matplotlib3",
],
)
+
+py_test(
+ name = "plot_test",
+ srcs = ["plot_test.py"],
+ data = [
+ "@sample_logfile//file",
+ ],
+ restricted_to = ["//tools:k8"],
+ deps = [":plot"],
+)
diff --git a/frc971/analysis/plot_test.py b/frc971/analysis/plot_test.py
new file mode 100644
index 0000000..1c57ae2
--- /dev/null
+++ b/frc971/analysis/plot_test.py
@@ -0,0 +1,23 @@
+#!/usr/bin/python3
+import unittest
+
+import matplotlib
+# Use a non-interactive backend so that the test can actually run...
+matplotlib.use('Agg')
+
+import frc971.analysis.plot
+
+
+class PlotterTest(unittest.TestCase):
+ def test_plotter(self):
+ """Basic test that makes sure that we can run the test without crashing."""
+ self.assertEqual(0,
+ frc971.analysis.plot.main([
+ "binary", "--logfile",
+ "external/sample_logfile/file/log.fbs",
+ "--config", "gyro.pb"
+ ]))
+
+
+if __name__ == '__main__':
+ unittest.main()