Correct tests re #191

invenia · Oct 21, 2020 · 3f69407 · 3f69407
1 parent 0d63462
commit 3f69407
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/sensitivities/functional/reducedim.jl b/test/sensitivities/functional/reducedim.jl
@@ -54,8 +54,8 @@
                          randn(rng, 10, 10, 10), randn(rng, 10, 10, 10))
 
         # Issue #123
-        x6_ = collect(1:10)
-        tens = (fill(10.0, (10,)), fill(10.0, (10, 1)))
+        x6_ = float.(1:10)
+        tens = (fill(10.0, (10,)), fill(10.0, (10,)))
         @test ∇(x->sum(sum(x, dims=2)))(x6_) == (oneslike(x6_),)
         @test ∇((x, y)->sum(sum(x, dims=2) .+ sum(y, dims=2)'))(x6_, x6_) == tens
         @test ∇((x, y)->sum(x .+ y'))(x6_, x6_) == tens