From f399179c022d7873ee1f63cb4a8db9c43317c033 Mon Sep 17 00:00:00 2001
From: Ali Hamdi <alha02@outlook.com>
Date: Fri, 9 Jun 2017 07:05:09 +0200
Subject: [PATCH] update tests

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

diff --git a/test/backend/tensorflow.jl b/test/backend/tensorflow.jl
index 8a2096c1..4cdb353b 100644
--- a/test/backend/tensorflow.jl
+++ b/test/backend/tensorflow.jl
@@ -24,27 +24,11 @@ test_anon(tf)
 end
 
 @testset "Ops" begin
-
-  @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
-
-  @testset "inv" begin
-    @net f(x) = inv(x)
-    m = tf(f)
-    @test m(A) ≈ inv(A)
-  end
-
-  @testset "det" begin
-    @net f(x) = det(x)
-    m = tf(f)
-    @test m(A) ≈ det(A)
-  end
-
+  A = convert(Array{Float32},randn(5,5))
+  u,s,v = tf(@net x -> svd(x))(A)
+  @test A ≈ u*diagm(s)*transpose(v)
+  @test tf(@net x -> inv(x))(A) ≈ inv(A)
+  @test tf(@net x -> det(x))(A) ≈ det(A)
 end
 
 end