TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

batching tests

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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]

1
test/runtests.jl
View File

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