Apply review suggestions by @murrayrm: revert comment change, remove parameter

bnavigator · bnavigator · commit 9a54254eec86 · 2021-03-20T00:55:50.000+01:00
diff --git a/control/tests/freqresp_test.py b/control/tests/freqresp_test.py
@@ -423,7 +423,7 @@ def test_dcgain_consistency():
     sys_ss = ctrl.tf2ss(sys_tf)
     assert 0 in sys_ss.pole()
 
-    # Finite (real) numerator over 0 denominator => inf
+    # Finite (real) numerator over 0 denominator => inf + nanj
     np.testing.assert_equal(
         sys_tf(0, warn_infinite=False), complex(np.inf, np.nan))
     np.testing.assert_equal(
@@ -433,9 +433,9 @@ def test_dcgain_consistency():
     np.testing.assert_equal(
         sys_ss(0j, warn_infinite=False), complex(np.inf, np.nan))
     np.testing.assert_equal(
-        sys_tf.dcgain(warn_infinite=False), np.inf)
+        sys_tf.dcgain(), np.inf)
     np.testing.assert_equal(
-        sys_ss.dcgain(warn_infinite=False), np.inf)
+        sys_ss.dcgain(), np.inf)
 
     # Set up transfer function with pole, zero at the origin
     sys_tf = ctrl.tf([1, 0], [1, 0])
@@ -448,7 +448,7 @@ def test_dcgain_consistency():
     np.testing.assert_equal(
         sys_tf(0j, warn_infinite=False), complex(np.nan, np.nan))
     np.testing.assert_equal(
-        sys_tf.dcgain(warn_infinite=False), np.nan)
+        sys_tf.dcgain(), np.nan)
 
     # Set up state space version
     sys_ss = ctrl.tf2ss(ctrl.tf([1, 0], [1, 1])) * \
@@ -462,15 +462,15 @@ def test_dcgain_consistency():
         np.testing.assert_equal(
             sys_ss(0j, warn_infinite=False), complex(np.nan, np.nan))
         np.testing.assert_equal(
-            sys_ss.dcgain(warn_infinite=False), np.nan)
+            sys_ss.dcgain(), np.nan)
     elif 0 in sys_ss.pole():
         # Pole at the origin, but zero elsewhere => should get (inf + nanj)
         np.testing.assert_equal(
             sys_ss(0, warn_infinite=False), complex(np.inf, np.nan))
         np.testing.assert_equal(
             sys_ss(0j, warn_infinite=False), complex(np.inf, np.nan))
         np.testing.assert_equal(
-            sys_ss.dcgain(warn_infinite=False), np.inf)
+            sys_ss.dcgain(), np.inf)
     else:
         # Near pole/zero cancellation => nothing sensible to check
         pass
