Update tests

JuliaFirstOrder · Dec 29, 2023 · c343469 · c343469
1 parent 18d139d
commit c343469
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -36,9 +36,10 @@ end
         @test is_generalized_quadratic(Zero())
 
         for T in [Float32, Float64]
-            @test let x = T[1.0, 2.0, 3.0]
-                prox(Zero(), x, T(42)) == (x, T(0))
-            end
+            x = T[1.0, 2.0, 3.0]
+            @test Zero()(x) == T(0)
+            @test prox(Zero(), x, T(42)) == (x, T(0))
+            @test prox(Zero(), x, ) == (x, T(0))
         end
 
     end
@@ -52,9 +53,11 @@ end
         @test is_generalized_quadratic(IndZero())
 
         for T in [Float32, Float64]
-            @test let x = T[1.0, 2.0, 3.0]
-                prox(IndZero(), x, T(42)) == (T[0, 0, 0], T(0))
-            end
+            x = T[1.0, 2.0, 3.0]
+            @test IndZero()(x) == T(Inf)
+            @test IndZero()(T[0, 0, 0]) == T(0)
+            @test prox(IndZero(), x, T(42)) == (T[0, 0, 0], T(0))
+            @test prox(IndZero(), x) == (T[0, 0, 0], T(0))
         end
 
     end