batching tests

This commit is contained in:
Mike J Innes 2017-02-20 19:34:57 +00:00
parent 8a071115b2
commit 28202792bc
2 changed files with 8 additions and 0 deletions

7
test/batching.jl Normal file
View File

@ -0,0 +1,7 @@
bs = Batch([[1,2,3],[4,5,6]])
@test typeof(bs) <: Batch
@test bs == [[1,2,3],[4,5,6]]
@test rawbatch(bs) == [1 2 3; 4 5 6]

View File

@ -13,6 +13,7 @@ macro tfonly(ex)
:(Base.find_in_path("TensorFlow") nothing && $(esc(ex)))
end
include("batching.jl")
include("basic.jl")
include("recurrent.jl")
include("backend.jl")