blob: ee6929d97e057e578ce5c1ab329d9ba4155e3ba0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
#!/usr/bin/python
import matplotlib
from test_thermal import BaseTestThermal
import cr2
class TestPIDController(BaseTestThermal):
def test_dataframe(self):
"""Test that PIDController() generates a valid data_frame"""
pid = cr2.Run().pid_controller
self.assertTrue(len(pid.data_frame) > 0)
self.assertTrue("err_integral" in pid.data_frame.columns)
self.assertEquals(pid.data_frame["err"].iloc[0], 1276)
def test_plot_controller(self):
"""Test PIDController.plot_controller()
As it happens with all plot functions, just test that it doesn't explode"""
pid = cr2.Run().pid_controller
pid.plot_controller()
matplotlib.pyplot.close('all')
pid.plot_controller(title="Antutu", width=20, height=5)
matplotlib.pyplot.close('all')
_, ax = matplotlib.pyplot.subplots()
pid.plot_controller(ax=ax)
matplotlib.pyplot.close('all')
|