Commit f1033d54 authored by Steven Cordwell's avatar Steven Cordwell
Browse files

check the shape of each sparse random matrix

parent 88203b68
......@@ -223,9 +223,13 @@ def test_exampleRand_dense_check():
def test_exampleRand_sparse_P_shape():
assert (len(Prs) == ACTIONS)
for a in range(ACTIONS):
assert (Prs[a].shape == (STATES, STATES))
def test_exampleRand_sparse_R_shape():
assert (len(Rrs) == ACTIONS)
for a in range(ACTIONS):
assert (Rrs[a].shape == (STATES, STATES))
def test_exampleRand_sparse_check():
assert mdp.check(Prs, Rrs) == None
......
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