Commit 940c3c72 authored by Steven Cordwell's avatar Steven Cordwell
Browse files

add tests for new ability to accept lists and tuples to check()

parent 107d1973
......@@ -76,6 +76,15 @@ def test_check_P_square_stochastic_nonnegative_object_sparse():
P[a] = speye(STATES, STATES).tocsr()
assert (check(P, R) == None)
# check: P - square, stochastic and non-negative lists
def test_check_P_square_stochastic_nonnegative_list_array():
P = []
R = rand(STATES, ACTIONS)
for a in xrange(ACTIONS):
P.append(eye(STATES))
assert (check(P, R) == None)
# check: R - square stochastic and non-negative sparse
def test_check_R_square_stochastic_nonnegative_sparse():
......
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