test deprecated 'k' keyword for rlocus

python-control · May 22, 2024 · b58f435 · b58f435
1 parent 206ca85
commit b58f435
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/control/tests/rlocus_test.py b/control/tests/rlocus_test.py
@@ -181,10 +181,11 @@ def test_root_locus_plots(sys, grid, xlim, ylim, interactive):
 
 
 # Test deprecated keywords
-def test_root_locus_legacy():
+@pytest.mark.parametrize("keyword", ["kvect", "k"])
+def test_root_locus_legacy(keyword):
     sys = ct.rss(2, 1, 1)
-    with pytest.warns(DeprecationWarning, match="'kvect' is deprecated"):
-        ct.root_locus_plot(sys, kvect=[0, 1, 2])
+    with pytest.warns(DeprecationWarning, match=f"'{keyword}' is deprecated"):
+        ct.root_locus_plot(sys, **{keyword: [0, 1, 2]})
 
 
 # Generate plots used in documentation