def test_include_check_partials_true_gl(self):
dm.options['include_check_partials'] = True
p = brachistochrone_min_time(transcription='gauss-lobatto',
compressed=False,
run_driver=False,
force_alloc_complex=True)
cpd = p.check_partials(out_stream=None, method='fd')
self.assertTrue(len(list(cpd.keys())) > 1)
def test_include_check_partials_false_radau(self):
dm.options['include_check_partials'] = False
p = brachistochrone_min_time(transcription='radau-ps',
compressed=False,
run_driver=False,
force_alloc_complex=True)
cpd = p.check_partials(out_stream=None)
self.assertSetEqual(set(cpd.keys()), {'traj0.phases.phase0.rhs_all'})
def test_ex_brachistochrone_gl_uncompressed(self):
ex_brachistochrone.SHOW_PLOTS = True
p = ex_brachistochrone.brachistochrone_min_time(transcription='gauss-lobatto',
compressed=False)
self.run_asserts(p)
self.tearDown()
if os.path.exists('ex_brach_gl_uncompressed.db'):
os.remove('ex_brach_gl_uncompressed.db')
def test_ex_brachistochrone_radau_compressed(self):
ex_brachistochrone.SHOW_PLOTS = True
p = ex_brachistochrone.brachistochrone_min_time(transcription='radau-ps',
compressed=True)
self.run_asserts(p)
self.tearDown()
if os.path.exists('ex_brach_radau_compressed.db'):
os.remove('ex_brach_radau_compressed.db')
def test_ex_brachistochrone_rk(self):
ex_brachistochrone.SHOW_PLOTS = True
p = ex_brachistochrone.brachistochrone_min_time(
transcription='runge-kutta')
self.run_asserts(p)
self.tearDown()
if os.path.exists('ex_brach_rk_uncompressed.db'):
os.remove('ex_brach_rk_uncompressed.db')
def test_include_check_partials_false_gl(self):
dm.options['include_check_partials'] = False
p = brachistochrone_min_time(transcription='gauss-lobatto',
compressed=False,
run_driver=False,
force_alloc_complex=True)
cpd = p.check_partials(out_stream=None, method='fd')
self.assertSetEqual(
set(cpd.keys()),
{'traj0.phases.phase0.rhs_disc', 'traj0.phases.phase0.rhs_col'})
def test_ex_brachistochrone_vs_gl_single_segment(self):
ex_brachistochrone_vs.SHOW_PLOTS = False
p = ex_brachistochrone.brachistochrone_min_time(
transcription='gauss-lobatto',
compressed=False,
force_alloc_complex=True,
solve_segments=True,
num_segments=1,
transcription_order=11)
self.assert_results(p)
def test_ex_brachistochrone_vs_radau_uncompressed(self):
ex_brachistochrone_vs.SHOW_PLOTS = False
p = ex_brachistochrone.brachistochrone_min_time(
transcription='radau-ps',
compressed=False,
force_alloc_complex=True,
solve_segments=True,
num_segments=10,
transcription_order=3)
self.assert_results(p)
def test_include_check_partials_false_rk(self):
dm.options['include_check_partials'] = False
p = brachistochrone_min_time(transcription='runge-kutta',
compressed=False,
run_driver=False,
force_alloc_complex=True)
cpd = p.check_partials(out_stream=None, method='fd')
self.assertSetEqual(
set(cpd.keys()), {
'traj0.phases.phase0.ode',
'traj0.phases.phase0.rk_solve_group.k_iter_group.ode'
})
