def test_simple_integration(self):
p = om.Problem(model=om.Group())
time_options = TimeOptionsDictionary()
time_options['units'] = 's'
time_options['targets'] = 't'
state_options = {}
state_options['y'] = StateOptionsDictionary()
state_options['y']['units'] = 'm'
state_options['y']['targets'] = 'y'
state_options['y']['rate_source'] = 'ydot'
gd = GridData(num_segments=4, transcription='gauss-lobatto', transcription_order=3)
seg0_comp = SegmentSimulationComp(index=0, grid_data=gd, method='RK45',
atol=1.0E-9, rtol=1.0E-9,
ode_class=TestODE, time_options=time_options,
state_options=state_options)
p.model.add_subsystem('segment_0', subsys=seg0_comp)
p.setup(check=True)
p.set_val('segment_0.time', [0, 0.25, 0.5])
p.set_val('segment_0.initial_states:y', 0.5)
p.run_model()
assert_rel_error(self,
p.get_val('segment_0.states:y', units='m')[-1, ...],
1.425639364649936,
tolerance=1.0E-6)
def test_simple_integration(self):
p = om.Problem(model=om.Group())
time_options = TimeOptionsDictionary()
time_options['units'] = 's'
time_options['targets'] = 't'
state_options = {}
state_options['y'] = StateOptionsDictionary()
state_options['y']['units'] = 'm'
state_options['y']['targets'] = 'y'
state_options['y']['rate_source'] = 'ydot'
# Non-standard way to assign state options, so we need this
state_options['y']['shape'] = (1, )
gd = GridData(num_segments=4, transcription='gauss-lobatto', transcription_order=3)
sim_options = SimulateOptionsDictionary()
sim_options['rtol'] = 1.0E-9
sim_options['atol'] = 1.0E-9
seg0_comp = SegmentSimulationComp(index=0, grid_data=gd, simulate_options=sim_options,
ode_class=TestODE, time_options=time_options,
state_options=state_options)
p.model.add_subsystem('segment_0', subsys=seg0_comp)
p.setup(check=True)
p.set_val('segment_0.time', [0, 0.25, 0.5])
p.set_val('segment_0.initial_states:y', 0.5)
p.run_model()
assert_near_equal(p.get_val('segment_0.states:y', units='m')[-1, ...],
1.425639364649936,
tolerance=1.0E-6)