Skip to content
Prev Previous commit
Merge branch 'master' into naive_bayes
  • Loading branch information
antmarakis authored Apr 14, 2017
commit 5e8e615de627dcacc52fb02607d1364705e03891
12 changes: 6 additions & 6 deletions tests/test_learning.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,15 +61,15 @@ def test_naive_bayes():

# Discrete
nBD = NaiveBayesLearner(iris)
assert nBD([5,3,1,0.1]) == "setosa"
assert nBD([6,5,3,1.5]) == "versicolor"
assert nBD([7,3,6.5,2]) == "virginica"
assert nBD([5, 3, 1, 0.1]) == "setosa"
assert nBD([6, 5, 3, 1.5]) == "versicolor"
assert nBD([7, 3, 6.5, 2]) == "virginica"

# Continuous
nBC = NaiveBayesLearner(iris, continuous=True)
assert nBC([5,3,1,0.1]) == "setosa"
assert nBC([6,5,3,1.5]) == "versicolor"
assert nBC([7,3,6.5,2]) == "virginica"
assert nBC([5, 3, 1, 0.1]) == "setosa"
assert nBC([6, 5, 3, 1.5]) == "versicolor"
assert nBC([7, 3, 6.5, 2]) == "virginica"


def test_k_nearest_neighbors():
Expand Down
You are viewing a condensed version of this merge commit. You can view the full changes here.