fix unit test precision in sisotool_test.py

murrayrm · murrayrm · commit e155a15e5330 · 2020-07-11T08:47:52.000-07:00
diff --git a/control/tests/sisotool_test.py b/control/tests/sisotool_test.py
@@ -33,8 +33,10 @@ def test_sisotool(self):
 
         # Check the step response before moving the point
         step_response_original = np.array(
-            [0.    , 0.0217, 0.1281, 0.3237, 0.5797, 0.8566, 1.116 , 1.3261, 1.4659, 1.526 ])
-        assert_array_almost_equal(ax_step.lines[0].get_data()[1][:10],step_response_original)
+            [0., 0.0217, 0.1281, 0.3237, 0.5797, 0.8566, 1.116,
+             1.3261, 1.4659, 1.526])
+        assert_array_almost_equal(
+            ax_step.lines[0].get_data()[1][:10], step_response_original, 4)
 
         bode_plot_params = {
             'omega': None,
@@ -76,8 +78,10 @@ def test_sisotool(self):
 
         # Check if the step response has changed
         step_response_moved = np.array(
-            [0.    , 0.0239, 0.161 , 0.4547, 0.8903, 1.407 , 1.9121, 2.2989, 2.4686, 2.353 ])
-        assert_array_almost_equal(ax_step.lines[0].get_data()[1][:10],step_response_moved)
+            [0., 0.0239, 0.161 , 0.4547, 0.8903, 1.407,
+             1.9121, 2.2989, 2.4686, 2.353])
+        assert_array_almost_equal(
+            ax_step.lines[0].get_data()[1][:10], step_response_moved, 4)
 
 
 if __name__ == "__main__":
