diff options
Diffstat (limited to 'tests/test_pid.py')
-rw-r--r-- | tests/test_pid.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/test_pid.py b/tests/test_pid.py index 8796f22..ce66d08 100644 --- a/tests/test_pid.py +++ b/tests/test_pid.py @@ -3,12 +3,12 @@ import matplotlib from test_thermal import BaseTestThermal -from cr2 import PIDController +import cr2 class TestPIDController(BaseTestThermal): def test_dataframe(self): """Test that PIDController() generates a valid data_frame""" - pid = PIDController() + pid = cr2.Run().pid_controller self.assertTrue(len(pid.data_frame) > 0) self.assertTrue("err_integral" in pid.data_frame.columns) @@ -18,7 +18,7 @@ class TestPIDController(BaseTestThermal): """Test PIDController.plot_controller() As it happens with all plot functions, just test that it doesn't explode""" - pid = PIDController() + pid = cr2.Run().pid_controller pid.plot_controller() matplotlib.pyplot.close('all') |