From 6a3f02b8e849e0f8313ebf77d474dfff6f62e468 Mon Sep 17 00:00:00 2001
From: Ali Hamdi <alha02@outlook.com>
Date: Thu, 8 Jun 2017 20:59:46 +0200
Subject: [PATCH] update tests

---
 test/backend/tensorflow.jl | 40 ++++++++++++++++++--------------------
 1 file changed, 19 insertions(+), 21 deletions(-)

diff --git a/test/backend/tensorflow.jl b/test/backend/tensorflow.jl
index 61f700f6..8a2096c1 100644
--- a/test/backend/tensorflow.jl
+++ b/test/backend/tensorflow.jl
@@ -25,28 +25,26 @@ end
 
 @testset "Ops" begin
 
-error_margin = 1e-4
+  @testset "svd" begin
+    A = convert(Array{Float32},randn(5,5))
+    @net f(x) = svd(x)
+    m = tf(f)
+    u,s,v = m(A)
+    @test A ≈ u*diagm(s)*transpose(v)
+  end
 
-#%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-# svd
-A = convert(Array{Float32},randn(5,5))
-@net f(x) = svd(x)
-m = tf(f)
-u,s,v = m(A)
-@test maximum(abs.(u*diagm(s)*transpose(v) - A)) < error_margin
+  @testset "inv" begin
+    @net f(x) = inv(x)
+    m = tf(f)
+    @test m(A) ≈ inv(A)
+  end
 
-#%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-# inv
-@net f(x) = inv(x)
-m = tf(f)
-@test maximum(abs.(m(A)-inv(A))) < error_margin
-
-#%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-# det
-@net f(x) = det(x)
-m = tf(f)
-@test maximum(abs.(m(A)-det(A))) < error_margin
-
-#%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+  @testset "det" begin
+    @net f(x) = det(x)
+    m = tf(f)
+    @test m(A) ≈ det(A)
+  end
+
+end
 
 end