Commit 263ea10c authored by Steven Cordwell's avatar Steven Cordwell
Browse files

dict tests for new ability to accept new types in check()

parent a04d5210
......@@ -99,6 +99,29 @@ def test_check_P_square_stochastic_nonnegative_list_sparse():
P.append(speye(STATES, STATES).tocsr())
assert (check(P, R) == None)
# check: P - square, stochastic and non-negative dicts
def test_check_P_square_stochastic_nonnegative_dict_array():
P = {}
R = rand(STATES, ACTIONS)
for a in xrange(ACTIONS):
P[a] = eye(STATES)
assert (check(P, R) == None)
def test_check_P_square_stochastic_nonnegative_dict_matrix():
P = []
R = rand(STATES, ACTIONS)
for a in xrange(ACTIONS):
P[a] = matrix(eye(STATES))
assert (check(P, R) == None)
def test_check_P_square_stochastic_nonnegative_dict_sparse():
P = []
R = rand(STATES, ACTIONS)
for a in xrange(ACTIONS):
P[a] = speye(STATES, STATES).tocsr()
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