add TapeTests

Author: jrevels

src/Tape.jl

@@ -27,6 +27,13 @@ end
 @inline capture(state::Tuple{Vararg{Number}}) = state
 @inline capture(state::Tuple) = map(capture, state)
 
+function Base.:(==)(a::TapeNode, b::TapeNode)
+    return (a.func == b.func &&
+            a.inputs == b.inputs &&
+            a.outputs == b.outputs &&
+            a.cache == b.cache)
+end
+
 ################################################
 # reverse pass (backpropagation over the tape) #
 ################################################

test/TapeTests.jl

@@ -0,0 +1,68 @@
+module TapeTests
+
+using ReverseDiffPrototype, Base.Test
+using ReverseDiffPrototype: TapeNode, Tape, Tracked
+
+const RDP = ReverseDiffPrototype
+
+include("utils.jl")
+
+println("testing Tape/TapeNode types...")
+tic()
+
+############################################################################################
+
+#################
+# TapeNode/Tape #
+#################
+
+x, y, k = [1, 2, 3], [4, 5, 6], 7
+z = x + y + k
+c = []
+tn = TapeNode(+, (x, y, k), z, c)
+@test tn.func === +
+@test tn.inputs === (x, y, k)
+@test tn.outputs === z
+@test tn.cache === c
+
+tp = Tape()
+ntp = Nullable(tp)
+RDP.record!(ntp, +, (x, y, k), z, c)
+tp1 = first(tp)
+@test tp1 == tn
+@test tp1.inputs[1] !== x
+@test tp1.inputs[2] !== y
+@test tp1.inputs[3] === k
+@test tp1.outputs !== z
+@test tp1.cache === c
+
+ntp = Nullable{Tape}()
+RDP.record!(ntp, +, (x, y, k), z, c)
+@test ntp === Nullable{Tape}()
+
+t = Tracked(1)
+x = [t, t]
+@test RDP.capture(t) === t
+
+cx = RDP.capture(x)
+@test cx !== x
+@test cx == x
+@test cx[1] === x[1]
+@test cx[2] === x[2]
+
+cs = RDP.capture((x, t, x))
+@test cs[1] !== x
+@test cs[1] == x
+@test cs[1][1] === x[1]
+@test cs[1][2] === x[2]
+@test cs[2] === t
+@test cs[3] !== x
+@test cs[3] == x
+@test cs[3][1] === x[1]
+@test cs[3][2] === x[2]
+
+############################################################################################
+
+println("done (took $(toq()) seconds)")
+
+end # module

test/TrackedTests.jl

@@ -1,6 +1,6 @@
 module TrackedTests
 
-using DiffBase, ForwardDiff, ReverseDiffPrototype, Base.Test
+using ReverseDiffPrototype, Base.Test
 using ReverseDiffPrototype: Tape, Tracked, value, adjoint, tape, valtype, adjtype
 
 const RDP = ReverseDiffPrototype

test/UtilsTests.jl

@@ -4,7 +4,7 @@ using DiffBase, ForwardDiff, ReverseDiffPrototype, Base.Test
 
 const RDP = ReverseDiffPrototype
 
-include("utils.jl")
+include("../utils.jl")
 
 println("testing gradient/gradient!...")
 tic()

test/GradientTests.jl renamed to test/api/GradientTests.jl

@@ -4,7 +4,7 @@ using DiffBase, ForwardDiff, ReverseDiffPrototype, Base.Test
 
 const RDP = ReverseDiffPrototype
 
-include("utils.jl")
+include("../utils.jl")
 
 println("testing hessian/hessian!...")
 tic()

test/HessianTests.jl renamed to test/api/HessianTests.jl

@@ -4,7 +4,7 @@ using DiffBase, ForwardDiff, ReverseDiffPrototype, Base.Test
 
 const RDP = ReverseDiffPrototype
 
-include("utils.jl")
+include("../utils.jl")
 
 println("testing jacobian/jacobian!...")
 tic()

test/JacobianTests.jl renamed to test/api/JacobianTests.jl

@@ -2,7 +2,8 @@ const TESTDIR = dirname(@__FILE__)
 
 testprintln(kind, f, pad = "  ") = println(pad, "testing $(kind): `$(f)`...")
 
+include(joinpath(TESTDIR, "TapeTests.jl"))
 include(joinpath(TESTDIR, "TrackedTests.jl"))
-include(joinpath(TESTDIR, "GradientTests.jl"))
-include(joinpath(TESTDIR, "JacobianTests.jl"))
-include(joinpath(TESTDIR, "HessianTests.jl"))
+include(joinpath(TESTDIR, "api/GradientTests.jl"))
+include(joinpath(TESTDIR, "api/JacobianTests.jl"))
+include(joinpath(TESTDIR, "api/HessianTests.jl"))