diff --git a/dymos/examples/brachistochrone/test/test_brachistochrone_control_rate_targets.py b/dymos/examples/brachistochrone/test/test_brachistochrone_control_rate_targets.py index be026f048..5e4d13146 100644 --- a/dymos/examples/brachistochrone/test/test_brachistochrone_control_rate_targets.py +++ b/dymos/examples/brachistochrone/test/test_brachistochrone_control_rate_targets.py @@ -126,6 +126,7 @@ def test_brachistochrone_control_rate_targets_gauss_lobatto(self): p.model.add_subsystem('phase0', phase) + phase.timeseries_options['include_control_rates'] = True phase.set_time_options(fix_initial=True, duration_bounds=(.5, 10)) phase.add_state('x', rate_source='xdot', @@ -411,6 +412,7 @@ def test_brachistochrone_control_rate_targets_radau(self): phase = dm.Phase(ode_class=BrachistochroneRateTargetODE, transcription=dm.Radau(num_segments=10)) + phase.timeseries_options['include_control_rates'] = True p.model.add_subsystem('phase0', phase) phase.set_time_options(fix_initial=True, duration_bounds=(.5, 10)) diff --git a/dymos/examples/brachistochrone/test/test_brachistochrone_vector_path_constraints.py b/dymos/examples/brachistochrone/test/test_brachistochrone_vector_path_constraints.py index 70e77ce5c..e9a84e6ff 100644 --- a/dymos/examples/brachistochrone/test/test_brachistochrone_vector_path_constraints.py +++ b/dymos/examples/brachistochrone/test/test_brachistochrone_vector_path_constraints.py @@ -32,6 +32,7 @@ def test_brachistochrone_vector_state_path_constraints_radau_partial_indices(sel p.model.add_subsystem('phase0', phase) + phase.timeseries_options['include_control_rates'] = True phase.set_time_options(fix_initial=True, duration_bounds=(.5, 10)) phase.add_state('pos', fix_initial=True, fix_final=True) @@ -144,6 +145,7 @@ def test_brachistochrone_vector_ode_path_constraints_radau_partial_indices(self) phase = dm.Phase(ode_class=BrachistochroneVectorStatesODE, transcription=dm.Radau(num_segments=20, order=3)) + phase.timeseries_options['include_control_rates'] = True p.model.add_subsystem('phase0', phase) phase.set_time_options(fix_initial=True, duration_bounds=(.5, 10)) @@ -257,6 +259,7 @@ def test_brachistochrone_vector_ode_path_constraints_radau_no_indices(self): transcription=dm.Radau(num_segments=20, order=3)) p.model.add_subsystem('phase0', phase) + phase.timeseries_options['include_control_rates'] = True phase.set_time_options(fix_initial=True, duration_bounds=(.5, 10)) @@ -369,6 +372,7 @@ def test_brachistochrone_vector_state_path_constraints_gl_partial_indices(self): phase = dm.Phase(ode_class=BrachistochroneVectorStatesODE, transcription=dm.GaussLobatto(num_segments=20, order=3)) + phase.timeseries_options['include_control_rates'] = True p.model.add_subsystem('phase0', phase) phase.set_time_options(fix_initial=True, duration_bounds=(.5, 10)) @@ -598,6 +602,7 @@ def test_brachistochrone_vector_ode_path_constraints_gl_no_indices(self): phase = dm.Phase(ode_class=BrachistochroneVectorStatesODE, transcription=dm.GaussLobatto(num_segments=20, order=3)) + phase.timeseries_options['include_control_rates'] = True p.model.add_subsystem('phase0', phase) phase.set_time_options(fix_initial=True, duration_bounds=(.5, 10))