add tests for Data.Iris module

2019-03-09 13:02:59 -08:00 · 2019-03-09 13:02:59 -08:00 · 0cac373539
parent f061df3d23
commit 0cac373539
1 changed files with 6 additions and 0 deletions
--- a/test/data.jl
+++ b/test/data.jl
@ -14,3 +14,9 @@ using Test
@test FashionMNIST.labels() isa Vector{Int64}

@test Data.Sentiment.train() isa Vector{Data.Tree{Any}}
+
+@test Iris.features() isa Matrix
+@test size(Iris.features()) == (4,150)
+
+@test Iris.labels() isa Vector{String}
+@test size(Iris.labels()) == (150,)