added test because why not

JuliaFirstOrder · Dec 2, 2018 · 8ab41c3 · 8ab41c3
1 parent 6d7aa35
commit 8ab41c3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/test_cubeNormL2.jl b/test/test_cubeNormL2.jl
@@ -14,7 +14,10 @@ for R in [Float16, Float32, Float64]
             predicates_test(f)
             x = randn(T, shape)
             call_test(f, x)
-            prox_test(f, x, R(0.5)+rand(R))
+            gamma = R(0.5)+rand(R)
+            y, f_y = prox_test(f, x, gamma)
+            grad_f_y, f_y = gradient(f, y)
+            @test grad_f_y ≈ x - y
         end
     end
 end