Source code for tests.test_utils
import numpy as np
[docs]
def value_in_array(result, expected, decimal=5):
diff = np.abs(expected - result)
assert np.any(diff < 1.0 / (10**decimal))
[docs]
def array_in_array(result, expected, decimal=5):
diff = np.abs(expected - result)
mask = diff < 1.0 / (10**decimal)
passing = np.all(mask, axis=1)
assert np.any(passing)