def test_return_number_as_ndarray_ndarray(value): """Turn number into a ndarray.""" arr = np.array(value) assert (return_as_ndarray(arr) == arr).all()
def test_return_number_as_ndarray_list_vals(value): """Turn number into a ndarray.""" assert (return_as_ndarray(value) == np.array(value)).all()