@@ -45,53 +45,52 @@ def test_stage(self, nsteps):
4545 stage_model .computeFirstOrderDerivatives (ep .x0 , ep .u0 , sd )
4646 stage_model .num_dual == ep .ndx
4747
48- def test_rollout (self , nsteps ):
49- import example_problem as ep
48+ # def test_rollout(self, nsteps):
49+ # import example_problem as ep
5050
51- us_i = [np .ones (ep .dyn_model .nu ) * 0.1 for _ in range (nsteps )]
52- xs_i = aligator .rollout (ep .dyn_model , ep .x0 , us_i ).tolist ()
53- dd = ep .dyn_model .createData ()
54- assert isinstance (dd , ep .TwistData )
55- ep .dyn_model .forward (ep .x0 , us_i [0 ], dd )
56- assert np .allclose (dd .xnext , xs_i [1 ])
51+ # us_i = [np.ones(ep.dyn_model.nu) * 0.1 for _ in range(nsteps)]
52+ # xs_i = aligator.rollout(ep.dyn_model, ep.x0, us_i).tolist()
53+ # dd = ep.dyn_model.createData()
54+ # assert isinstance(dd, ep.TwistData)
55+ # ep.dyn_model.forward(ep.x0, us_i[0], dd)
5756
58- def test_shooting_problem (self , nsteps ):
59- import example_problem as ep
57+ # def test_shooting_problem(self, nsteps):
58+ # import example_problem as ep
6059
61- stage_model = ep .stage_model
62- problem = aligator .TrajOptProblem (ep .x0 , ep .nu , ep .space , term_cost = ep .cost )
63- for _ in range (nsteps ):
64- problem .addStage (stage_model )
60+ # stage_model = ep.stage_model
61+ # problem = aligator.TrajOptProblem(ep.x0, ep.nu, ep.space, term_cost=ep.cost)
62+ # for _ in range(nsteps):
63+ # problem.addStage(stage_model)
6564
66- problem_data = aligator .TrajOptData (problem )
65+ # problem_data = aligator.TrajOptData(problem)
6766
68- print ("term cost data:" , problem_data .term_cost )
69- print ("term cstr data:" , problem_data .term_constraint )
67+ # print("term cost data:", problem_data.term_cost)
68+ # print("term cstr data:", problem_data.term_constraint)
7069
71- stage2 = stage_model .copy ()
72- sd0 = stage2 .createData ()
73- print ("Clone stage:" , stage2 )
74- print ("Clone stage data:" , sd0 )
70+ # stage2 = stage_model.copy()
71+ # sd0 = stage2.createData()
72+ # print("Clone stage:", stage2)
73+ # print("Clone stage data:", sd0)
7574
76- us_init = [ep .u0 ] * nsteps
77- xs_out = aligator .rollout (ep .dyn_model , ep .x0 , us_init ).tolist ()
75+ # us_init = [ep.u0] * nsteps
76+ # xs_out = aligator.rollout(ep.dyn_model, ep.x0, us_init).tolist()
7877
79- assert len (problem_data .stage_data ) == problem .num_steps
80- assert problem .num_steps == nsteps
78+ # assert len(problem_data.stage_data) == problem.num_steps
79+ # assert problem.num_steps == nsteps
8180
82- problem .evaluate (xs_out , us_init , problem_data )
83- problem .computeDerivatives (xs_out , us_init , problem_data )
81+ # problem.evaluate(xs_out, us_init, problem_data)
82+ # problem.computeDerivatives(xs_out, us_init, problem_data)
8483
85- solver = ep .solver
84+ # solver = ep.solver
8685
87- assert solver .bcl_params .prim_alpha == 0.1
88- assert solver .bcl_params .prim_beta == 0.9
89- assert solver .bcl_params .dual_alpha == 1.0
90- assert solver .bcl_params .dual_beta == 1.0
86+ # assert solver.bcl_params.prim_alpha == 0.1
87+ # assert solver.bcl_params.prim_beta == 0.9
88+ # assert solver.bcl_params.dual_alpha == 1.0
89+ # assert solver.bcl_params.dual_beta == 1.0
9190
92- solver .setup (problem )
93- solver .rollout_type = aligator .ROLLOUT_LINEAR
94- solver .run (problem , xs_out , us_init )
91+ # solver.setup(problem)
92+ # solver.rollout_type = aligator.ROLLOUT_LINEAR
93+ # solver.run(problem, xs_out, us_init)
9594
9695
9796if __name__ == "__main__" :
0 commit comments