more cleanup

OpenMDAO · Nov 14, 2024 · 3153fe4 · 3153fe4
1 parent 4085266
commit 3153fe4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/dymos/transcriptions/explicit_shooting/test/test_explicit_shooting.py b/dymos/transcriptions/explicit_shooting/test/test_explicit_shooting.py
@@ -226,10 +226,14 @@ def test_brachistochrone_explicit_shooting(self):
                 assert_near_equal(t[-1, ...], 1.8016, tolerance=1.0E-2)
 
                 with np.printoptions(linewidth=1024):
-                    cpd = prob.check_partials(compact_print=True, method='cs', excludes=['traj0.phases.phase0.integrator'], out_stream=None)
+                    cpd = prob.check_partials(compact_print=True, method='cs',
+                                              excludes=['traj0.phases.phase0.integrator'],
+                                              out_stream=None)
                     assert_check_partials(cpd, atol=1.0E-5, rtol=1.0E-5)
 
-                    cpd = prob.check_partials(compact_print=True, method='fd', includes=['traj0.phases.phase0.integrator'], out_stream=None)
+                    cpd = prob.check_partials(compact_print=True, method='fd',
+                                              includes=['traj0.phases.phase0.integrator'],
+                                              out_stream=None)
                     assert_check_partials(cpd, atol=1.0E-5, rtol=1.0E-5)
 
         dymos_options['include_check_partials'] = False

diff --git a/dymos/transcriptions/pseudospectral/components/test/test_collocation_defect_sol_opt.py b/dymos/transcriptions/pseudospectral/components/test/test_collocation_defect_sol_opt.py
@@ -191,6 +191,7 @@ def test_partials(self):
 
         p = self.make_prob('radau-ps', n_segs=2, order=5, compressed=True)
         cpd = p.check_partials(compact_print=True, method='fd', out_stream=None)
+        del cpd['state_indep']
         assert_check_partials(cpd)
 
         p = self.make_prob('gauss-lobatto', n_segs=3, order=5, compressed=False)