From cab09e1c8ce494eb610851adfd7b7fc7bf711258 Mon Sep 17 00:00:00 2001 From: Antonis Maronikolakis Date: Fri, 7 Apr 2017 12:31:13 +0300 Subject: [PATCH] Update test_learning.py --- tests/test_learning.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/test_learning.py b/tests/test_learning.py index 4f618f7c1..60ea61366 100644 --- a/tests/test_learning.py +++ b/tests/test_learning.py @@ -41,6 +41,8 @@ def test_k_nearest_neighbors(): kNN = NearestNeighborLearner(iris,k=3) assert kNN([5,3,1,0.1]) == "setosa" + assert kNN([6,5,3,1.5]) == "versicolor" + assert kNN([7.5,4,6,2]) == "virginica" def test_decision_tree_learner(): @@ -48,6 +50,8 @@ def test_decision_tree_learner(): dTL = DecisionTreeLearner(iris) assert dTL([5,3,1,0.1]) == "setosa" + assert dTL([6,5,3,1.5]) == "versicolor" + assert dTL([7.5,4,6,2]) == "virginica" def test_neural_network_learner():