Remove imports needed for eval

roryyorke · roryyorke · commit 1bc01197004d · 2025-02-16T10:31:27.000+02:00
Provide relevant symbols via `locals` argument to eval.
diff --git a/control/tests/config_test.py b/control/tests/config_test.py
@@ -339,11 +339,8 @@ def test_system_indexing(self):
         {'dt': 0.1}
     ])
     def test_repr_format(self, kwargs):
-        from ..statesp import StateSpace
-        from numpy import array
-
         sys = ct.ss([[1]], [[1]], [[1]], [[0]], **kwargs)
-        new = eval(repr(sys))
+        new = eval(repr(sys), locals={'StateSpace':ct.StateSpace, 'array':np.array})
         for attr in ['A', 'B', 'C', 'D']:
             assert getattr(new, attr) == getattr(sys, attr)
         for prop in ['input_labels', 'output_labels', 'state_labels']:
diff --git a/control/tests/namedio_test.py b/control/tests/namedio_test.py
@@ -366,15 +366,15 @@ def test_negative_system_spec():
 
 # Named signal representations
 def test_named_signal_repr():
-    from numpy import array
-    from ..iosys import NamedSignal
     sys = ct.rss(
         states=2, inputs=['u1', 'u2'], outputs=['y1', 'y2'],
         state_prefix='xi')
     resp = sys.step_response(np.linspace(0, 1, 3))
 
     for signal in ['inputs', 'outputs', 'states']:
         sig_orig = getattr(resp, signal)
-        sig_eval = eval(repr(sig_orig))
+        sig_eval = eval(repr(sig_orig),
+                        locals={'array': np.array,
+                                'NamedSignal': ct.NamedSignal})
         assert sig_eval.signal_labels == sig_orig.signal_labels
         assert sig_eval.trace_labels == sig_orig.trace_labels
