test to make sure kvect gains are respected even if plotting

sawyerbfuller · sawyerbfuller · commit eadd496bf581 · 2022-12-09T14:55:56.000-08:00
diff --git a/control/tests/rlocus_test.py b/control/tests/rlocus_test.py
@@ -54,6 +54,12 @@ def testRootLocus(self, sys):
         np.testing.assert_allclose(klist, k_out)
         self.check_cl_poles(sys, roots, klist)
 
+        # now check with plotting
+        roots, k_out = root_locus(sys, klist)
+        np.testing.assert_equal(len(roots), len(klist))
+        np.testing.assert_allclose(klist, k_out)
+        self.check_cl_poles(sys, roots, klist)
+
     def test_without_gains(self, sys):
         roots, kvect = root_locus(sys, plot=False)
         self.check_cl_poles(sys, roots, kvect)
