Add tests for generate checkerboard

ATISLabs · Sep 6, 2020 · 66dc4e6 · 66dc4e6
1 parent 60a88e2
commit 66dc4e6
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -48,4 +48,16 @@ using Test
     @test size(data)[1] == samples
     @test size(data)[2] == features + 1
 
+    n_clusters = 3
+    samples = 10
+    features = 4
+    X, rows, cols = SyntheticDatasets.generate_checkerboard((samples, features), n_clusters)
+
+    @test size(X)[1] == samples
+    @test size(X)[2] == features
+
+    @test size(rows)[2] == samples 
+
+    @test size(cols)[2] == features
+
 end