unit tests now test for deprecation of sys.freqresp

sawyerbfuller · sawyerbfuller · commit 330e51700211 · 2020-08-19T10:44:32.000-07:00
diff --git a/control/tests/statesp_test.py b/control/tests/statesp_test.py
@@ -223,7 +223,7 @@ def test_freqresp_deprecated(self):
 
             # Make sure that we get a pending deprecation warning
             sys.freqresp(1.)
-            assert issubclass(w[-1].category, PendingDeprecationWarning)
+            assert issubclass(w[-1].category, DeprecationWarning)
 
     @unittest.skipIf(not slycot_check(), "slycot not installed")
     def test_freq_resp(self):
diff --git a/control/tests/xferfcn_test.py b/control/tests/xferfcn_test.py
@@ -432,6 +432,19 @@ def test_call_mimo(self):
         # Test call version as well
         np.testing.assert_array_almost_equal(sys(2.j), resp)
 
+    def test_freqresp_deprecated(self):
+        sys = TransferFunction([1., 3., 5], [1., 6., 2., -1.])
+        # Deprecated version of the call (should generate warning)
+        import warnings
+        with warnings.catch_warnings(record=True) as w:
+            # Set up warnings filter to only show warnings in control module
+            warnings.filterwarnings("ignore")
+            warnings.filterwarnings("always", module="control")
+
+            # Make sure that we get a pending deprecation warning
+            sys.freqresp(1.)
+            assert issubclass(w[-1].category, DeprecationWarning)
+
     def test_frequency_response_siso(self):
         """Evaluate the magnitude and phase of a SISO system at
         multiple frequencies."""
