def assert_allclose(lhs, rhs): for tb in map(zip, lhs, rhs): for left, right in tb: if isinstance(left, (np.floating, float)): # account for nans assert np.all(np.isclose(left, right, equal_nan=True)) else: assert left == right
def create_index(t, columns, name=None, **kwargs): if not all(map(partial(hasattr, t.cols), columns)): raise ValueError('table %s does not have all passed in columns %s' % (t, columns)) for column in columns: create_index(t, column, **kwargs)