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' })