Commit 553cce46 authored by Steven Cordwell's avatar Steven Cordwell
Browse files

fix for PolicyIteration's policy0 code unit test

parent 2e499bdb
......@@ -230,7 +230,7 @@ def test_ValueIteration_exampleForest():
def test_PolicyIteration_init_policy0():
a = PolicyIteration(P, R, 0.9)
p = array((1, 1)).reshape(2, 1)
assert (a.policy - p < SMALLNUM).all()
assert (absolute(a.policy - p) < SMALLNUM).all()
def test_PolicyIteration():
PolicyIteration(P, R, 0.9)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment