Set time symbol everywhere.

csu-hmc · Jul 15, 2024 · 37caeec · 37caeec
1 parent d7bdd4e
commit 37caeec
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/opty/tests/test_direct_collocation.py b/opty/tests/test_direct_collocation.py
@@ -80,6 +80,7 @@ def test_Problem():
                    state_symbols,
                    2,
                    interval_value,
+                   time_symbol=t,
                    bounds={x: (-10.0, 10.0),
                            f: (-8.0, 8.0),
                            m: (-1.0, 1.0),
@@ -133,7 +134,8 @@ def setup_method(self):
                                  num_collocation_nodes=4,
                                  node_time_interval=self.interval_value,
                                  known_parameter_map=par_map,
-                                 known_trajectory_map=traj_map)
+                                 known_trajectory_map=traj_map,
+                                 time_symbol=t)
 
     def test_init(self):
 
@@ -866,7 +868,8 @@ def setup_method(self):
                                  num_collocation_nodes=4,
                                  node_time_interval=self.interval_value,
                                  known_parameter_map=par_map,
-                                 instance_constraints=instance_constraints)
+                                 instance_constraints=instance_constraints,
+                                 time_symbol=t)
 
     def test_init(self):