Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit 3a0de56

Browse files
Chipe1norvig
authored andcommitted
Added seed for random test (aimacode#608)
1 parent a452213 commit 3a0de56

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/test_learning.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,14 +152,16 @@ def test_decision_tree_learner():
152152

153153

154154
def test_random_forest():
155+
random.seed("aima-python")
155156
iris = DataSet(name="iris")
156157
rF = RandomForest(iris)
157158
assert rF([5, 3, 1, 0.1]) == "setosa"
158-
assert rF([6, 5, 3, 1.5]) == "versicolor"
159+
assert rF([6, 5, 3, 1]) == "versicolor"
159160
assert rF([7.5, 4, 6, 2]) == "virginica"
160161

161162

162163
def test_neural_network_learner():
164+
random.seed("aima-python")
163165
iris = DataSet(name="iris")
164166
classes = ["setosa", "versicolor", "virginica"]
165167
iris.classes_to_numbers(classes)

0 commit comments

Comments
 (0)