test: fix SII tests

SciML · Nov 13, 2024 · ec79535 · ec79535
1 parent d837b1f
commit ec79535
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test/mtk_cache_indexing_tests.jl b/test/mtk_cache_indexing_tests.jl
@@ -1,6 +1,7 @@
 @testitem "Modeling Toolkit Cache Indexing" tags=[:downstream] begin
     using ModelingToolkit
     using ModelingToolkit: t_nounits as t
+    import NonlinearSolveBase, NonlinearSolveFirstOrder
 
     @parameters p d
     @variables X(t)
@@ -11,9 +12,9 @@
     nlprob = NonlinearProblem(nlsys, [X => 1.0], [p => 2.0, d => 3.0])
 
     @testset "$integtype" for (alg, integtype) in [
-        (NewtonRaphson(), NonlinearSolve.GeneralizedFirstOrderAlgorithmCache),
-        (FastShortcutNonlinearPolyalg(), NonlinearSolve.NonlinearSolvePolyAlgorithmCache),
-        (SimpleNewtonRaphson(), NonlinearSolve.NonlinearSolveNoInitCache)
+        (NewtonRaphson(), NonlinearSolveFirstOder.GeneralizedFirstOrderAlgorithmCache),
+        (FastShortcutNonlinearPolyalg(), NonlinearSolveBase.NonlinearSolvePolyAlgorithmCache),
+        (SimpleNewtonRaphson(), NonlinearSolveBase.NonlinearSolveNoInitCache)
     ]
         nint = init(nlprob, alg)
         @test nint isa integtype