Commit 963ad866 authored by Steven Cordwell's avatar Steven Cordwell
Browse files

[tests] Use mdptoolbox.example.small in the tests

parent 6b126aec
...@@ -15,12 +15,11 @@ ACTIONS = 3 ...@@ -15,12 +15,11 @@ ACTIONS = 3
SMALLNUM = 10e-12 SMALLNUM = 10e-12
# np.arrays # np.arrays
P_small = np.array([[[0.5, 0.5],[0.8, 0.2]],[[0, 1],[0.1, 0.9]]]) P_small, R_small = mdptoolbox.example.small()
R_small = np.array([[5, 10], [-1, 2]])
P_sparse = np.empty(2, dtype=object) P_sparse = np.empty(2, dtype=object)
P_sparse[0] = sp.sparse.csr_matrix([[0.5, 0.5],[0.8, 0.2]]) P_sparse[0] = sp.sparse.csr_matrix(P_small[0])
P_sparse[1] = sp.sparse.csr_matrix([[0, 1],[0.1, 0.9]]) P_sparse[1] = sp.sparse.csr_matrix(P_small[1])
P_forest, R_forest = mdptoolbox.example.forest() P_forest, R_forest = mdptoolbox.example.forest()
......
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