@@ -258,7 +258,7 @@ def test_combine_time_responses():
258
258
sys_mimo = ct .rss (4 , 2 , 2 )
259
259
timepts = np .linspace (0 , 10 , 100 )
260
260
261
- # Combine two response with ntrace = 0
261
+ # Combine two responses with ntrace = 0
262
262
U = np .vstack ([np .sin (timepts ), np .cos (2 * timepts )])
263
263
resp1 = ct .input_output_response (sys_mimo , timepts , U )
264
264
@@ -293,6 +293,7 @@ def test_combine_time_responses():
293
293
combresp4 = ct .combine_time_responses (
294
294
[resp1 , resp2 , resp3 ], trace_labels = labels )
295
295
assert combresp4 .trace_labels == labels
296
+ assert combresp4 .trace_types == [None , None , 'step' , 'step' ]
296
297
297
298
# Automatically generated trace label names and types
298
299
resp5 = ct .step_response (sys_mimo , timepts )
@@ -302,7 +303,13 @@ def test_combine_time_responses():
302
303
combresp5 = ct .combine_time_responses ([resp1 , resp5 ])
303
304
assert combresp5 .trace_labels == [resp1 .title ] + \
304
305
["test, trace 0" , "test, trace 1" ]
305
- assert combresp4 .trace_types == [None , None , 'step' , 'step' ]
306
+ assert combresp5 .trace_types == [None , None , None ]
307
+
308
+ # ntraces = 0 with trace_types != None
309
+ # https://github.com/python-control/python-control/issues/1025
310
+ resp6 = ct .forced_response (sys_mimo , timepts , U )
311
+ combresp6 = ct .combine_time_responses ([resp1 , resp6 ])
312
+ assert combresp6 .trace_types == [None , 'forced' ]
306
313
307
314
with pytest .raises (ValueError , match = "must have the same number" ):
308
315
resp = ct .step_response (ct .rss (4 , 2 , 3 ), timepts )
0 commit comments