add tests for utilities

FluxML · Jan 21, 2024 · 80cd5fa · 80cd5fa
1 parent fa7dc1f
commit 80cd5fa
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/test/ext_cuda/runtests.jl b/test/ext_cuda/runtests.jl
@@ -32,3 +32,7 @@ end
 @testset "ctc" begin
   include("ctc.jl")
 end
+@testset "utils" begin
+  include("utils.jl")
+end
+
diff --git a/test/ext_cuda/utils.jl b/test/ext_cuda/utils.jl
@@ -9,3 +9,9 @@
     @test y isa Wrapped{<:CuArray}
   end
 end
+
+@testset "rng_from_array" begin
+    x = cu(randn(2,2))
+    rng = Flux.rng_from_array(x)
+    @test rng == CUDA.default_rng()
+end