aboutsummaryrefslogtreecommitdiff
path: root/tests/test_pid.py
diff options
context:
space:
mode:
authorJavi Merino <javi.merino@arm.com>2014-08-14 15:32:36 +0100
committerJavi Merino <javi.merino@arm.com>2015-08-12 14:52:42 +0100
commit9d6dabdd9eb28f75988e9f8bb98ef6f65ef7d995 (patch)
tree2de77d55d13d98297d9e91b8842242edf3cf852a /tests/test_pid.py
parent903d2c0508986886f4160d20bc6190aa6db466ab (diff)
downloadtrappy-9d6dabdd9eb28f75988e9f8bb98ef6f65ef7d995.tar.gz
tests: make them get the trace classes from explicitly from Run() invocations
Trace classes are about to be removed from the directly exposed module classes so don't make tests rely on it. Move a couple of checks from ThermalGovernor tests to the Run() tests as it's testing basic functionality that really belongs to the Run() class now.
Diffstat (limited to 'tests/test_pid.py')
-rw-r--r--tests/test_pid.py6
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')