Maky identity size dependent on inputs

bnavigator · bnavigator · commit 24af4fde5041 · 2025-02-01T11:45:30.000+01:00
diff --git a/control/tests/statesp_test.py b/control/tests/statesp_test.py
@@ -570,7 +570,7 @@ def test_pow(self, request, sysname, power):
         sys = request.getfixturevalue(sysname)
         result = sys**power
         if power == 0:
-             expected = StateSpace([], [], [], np.eye(2), dt=0)
+             expected = StateSpace([], [], [], np.eye(sys.ninputs), dt=0)
         else:
             sign = 1 if power > 0 else -1
             expected = sys**sign
@@ -597,7 +597,7 @@ def test_pow_inv(self, request, sysname, order):
             result = (sys**-1 * sys).minreal()
         else:
             result = (sys * sys**-1).minreal()
-        expected = StateSpace([], [], [], np.eye(2), dt=0)
+        expected = StateSpace([], [], [], np.eye(sys.ninputs), dt=0)
         assert_tf_close_coeff(
             ss2tf(expected).minreal(),
             ss2tf(result).minreal())
