Commit 0970cc4a authored by Steven Cordwell's avatar Steven Cordwell
Browse files

[tests] Fix doctests so tht most will pass

parent 3711c2da
This diff is collapsed.
...@@ -114,6 +114,9 @@ def check(P, R): ...@@ -114,6 +114,9 @@ def check(P, R):
>>> import numpy as np >>> import numpy as np
>>> P_invalid = np.random.rand(5, 100, 100) >>> P_invalid = np.random.rand(5, 100, 100)
>>> mdptoolbox.util.check(P_invalid, R_valid) # Raises an exception >>> mdptoolbox.util.check(P_invalid, R_valid) # Raises an exception
Traceback (most recent call last):
...
AssertionError: Each row of a transition probability matrix must sum to one (1).
""" """
# Checking P # Checking P
...@@ -295,7 +298,7 @@ def checkSquareStochastic(Z): ...@@ -295,7 +298,7 @@ def checkSquareStochastic(Z):
raise InvalidMDPError(mdperr["mat_square"]) raise InvalidMDPError(mdperr["mat_square"])
# check that the matrix is square, and that each row sums to one # check that the matrix is square, and that each row sums to one
assert s1 == s2, mdperr["mat_square"] assert s1 == s2, mdperr["mat_square"]
assert (_np.abs(Z.sum(axis=1) - _np.ones(s2))).max() <= _np.spacing(1), \ assert (_np.abs(Z.sum(axis=1) - _np.ones(s2))).max() <= 2*_np.spacing(1), \
mdperr["mat_stoch"] mdperr["mat_stoch"]
# make sure that there are no values less than zero # make sure that there are no values less than zero
try: try:
......
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