Added tests

Fix tests gaffe
FluxML · Feb 17, 2022 · 4534056 · 4534056
1 parent 393ef68
commit 4534056
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/test/utils.jl b/test/utils.jl
@@ -146,6 +146,15 @@ end
     end
   end
 
+  @testset "truncated_normal" begin
+    for sz in [(100,), (100, 100), (2, 3, 32, 64), (2, 3, 4, 32, 64)]
+      v = truncated_normal(sz...)
+      @test -1.0 < minimum(v) < 0.0
+      @test 0.0 < maximum(v) < 1.0
+      @test eltype(v) == Float32
+    end
+  end
+
   @testset "partial_application" begin
     big = 1e9