Tests: make external initialization deterministic (#75)

JuliaStats · Jan 26, 2023 · e319cd8 · e319cd8
1 parent e2139ac
commit e319cd8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/initialization.jl b/test/initialization.jl
@@ -42,11 +42,11 @@
         @test all(W2 .== W)
         @test all(H2 .== zero(T))
 
-        Random.seed!(5678)
-        W1, H1 = NMF.nndsvd(X, 5);
         F = svd(X)
-        W2, H2 = NMF.nndsvd(X, 5; initdata=F)
-        @test norm(X - W2*H2) < norm(X - W1*H1)
+        W1, H1 = NMF.nndsvd(X, 5; initdata=F)
+        F = svd(2X)
+        W2, H2 = NMF.nndsvd(2X, 5; initdata=F)
+        @test W2 ≈ sqrt(T(2)) * W1 && H2 ≈ sqrt(T(2)) * H1
 
         W, H = NMF.nndsvd(X, 5; variant=:ar)
         @test all(W .> zero(T))