|
| 1 | +importpytest |
| 2 | +importinspect |
| 3 | +importnumpyasnp |
| 4 | +importcontrolasct |
| 5 | + |
| 6 | +@pytest.mark.parametrize( |
| 7 | +"dt1, dt2, dt3", [ |
| 8 | + (0,0,0), |
| 9 | + (0,0.1,ValueError), |
| 10 | + (0,None,0), |
| 11 | + (0,True,ValueError), |
| 12 | + (0.1,0,ValueError), |
| 13 | + (0.1,0.1,0.1), |
| 14 | + (0.1,None,0.1), |
| 15 | + (0.1,True,0.1), |
| 16 | + (None,0,0), |
| 17 | + (None,0.1,0.1), |
| 18 | + (None,None,None), |
| 19 | + (None,True,True), |
| 20 | + (True,0,ValueError), |
| 21 | + (True,0.1,0.1), |
| 22 | + (True,None,True), |
| 23 | + (True,True,True), |
| 24 | + (0.2,None,0.2), |
| 25 | + (0.2,0.1,ValueError), |
| 26 | + ]) |
| 27 | +@pytest.mark.parametrize("op", [ct.series,ct.parallel,ct.feedback]) |
| 28 | +@pytest.mark.parametrize("type", [ct.StateSpace,ct.ss,ct.tf]) |
| 29 | +deftest_composition(dt1,dt2,dt3,op,type): |
| 30 | +# Define the system |
| 31 | +A,B,C,D= [[1,1], [0,1]], [[0], [1]], [[1,0]],0 |
| 32 | +sys1=ct.StateSpace(A,B,C,D,dt1) |
| 33 | +sys2=ct.StateSpace(A,B,C,D,dt2) |
| 34 | + |
| 35 | +# Convert to the desired form |
| 36 | +sys1=type(sys1) |
| 37 | +sys2=type(sys2) |
| 38 | + |
| 39 | +ifinspect.isclass(dt3)andissubclass(dt3,Exception): |
| 40 | +withpytest.raises(dt3,match="incompatible timebases"): |
| 41 | +sys3=op(sys1,sys2) |
| 42 | +else: |
| 43 | +sys3=op(sys1,sys2) |
| 44 | +assertsys3.dt==dt3 |
| 45 | + |
| 46 | + |
| 47 | +@pytest.mark.parametrize("dt", [None,0,0.1]) |
| 48 | +deftest_composition_override(dt): |
| 49 | +# Define the system |
| 50 | +A,B,C,D= [[1,1], [0,1]], [[0], [1]], [[1,0]],0 |
| 51 | +sys1=ct.ss(A,B,C,D,None,inputs='u1',outputs='y1') |
| 52 | +sys2=ct.ss(A,B,C,D,None,inputs='y1',outputs='y2') |
| 53 | + |
| 54 | +# Show that we can override the type |
| 55 | +sys3=ct.interconnect([sys1,sys2],inputs='u1',outputs='y2',dt=dt) |
| 56 | +assertsys3.dt==dt |
| 57 | + |
| 58 | +# Overriding the type with an inconsistent type generates an error |
| 59 | +sys1=ct.StateSpace(A,B,C,D,0.1,inputs='u1',outputs='y1') |
| 60 | +ifdt!=0.1anddtisnotNone: |
| 61 | +withpytest.raises(ValueError,match="incompatible timebases"): |
| 62 | +sys3=ct.interconnect( |
| 63 | + [sys1,sys2],inputs='u1',outputs='y2',dt=dt) |
| 64 | + |
| 65 | +sys1=ct.StateSpace(A,B,C,D,0,inputs='u1',outputs='y1') |
| 66 | +ifdt!=0anddtisnotNone: |
| 67 | +withpytest.raises(ValueError,match="incompatible timebases"): |
| 68 | +sys3=ct.interconnect( |
| 69 | + [sys1,sys2],inputs='u1',outputs='y2',dt=dt) |