using TensorFlow
Flux.loadtf()

@testset "TensorFlow" begin

xs, ys = rand(1, 20), rand(1, 20)
d = Affine(20, 10)

dt = tf(d)
@test d(xs) ≈ dt(xs)

@testset "Tuple I/O" begin
  @test tf(@net x -> (identity(x),))([1,2,3]) == ([1,2,3],)
  @test tf(@net x -> x[1].*x[2])(([1,2,3],[4,5,6])) == [4,10,18]
end

@testset "Recurrence" begin
  seq = batchone(Seq(rand(10) for i = 1:3))
  r = unroll(Recurrent(10, 5), 3)
  rm = tf(r)
  @test r(seq) ≈ rm(seq)
end

@testset "Tensor interface" begin
  sess = TensorFlow.Session()
  X = placeholder(Float32)
  Y = Tensor(d, X)
  run(sess, global_variables_initializer())

  @test run(sess, Y, Dict(X=>Float32.(xs))) ≈ d(xs)
end

@testset "Stack Traces" begin
  model = TLP(Affine(10, 20), Affine(21, 15))
  dm = tf(model)
  e = try dm(rand(1, 10))
  catch e e end

  @test isa(e, DataFlow.Interpreter.Exception)
  @test e.trace[1].func == Symbol("Flux.Affine")
  @test e.trace[2].func == :TLP
end

end