def test_within_range():
df = pd.DataFrame({'A': [-1, 0, 1]})
items = {'A': (-1, 1)}
tm.assert_frame_equal(df, ck.within_range(df, items))
tm.assert_frame_equal(df, dc.within_range(items)(_noop)(df))
items['A'] = (0, 1)
with pytest.raises(AssertionError):
ck.within_range(df, items)
with pytest.raises(AssertionError):
dc.within_range(items)(_noop)(df)
def test_within_range():
df = pd.DataFrame({'A': [-1, 0, 1]})
items = {'A': (-1, 1)}
tm.assert_frame_equal(df, ck.within_range(df, items))
tm.assert_frame_equal(df, dc.within_range(items)(_noop)(df))
items['A'] = (0, 1)
with pytest.raises(AssertionError):
ck.within_range(df, items)
with pytest.raises(AssertionError):
dc.within_range(items)(_noop)(df)
def wrapper(*args, **kwargs):
result = func(*args, **kwargs)
ck.within_range(result, items)
return result
def check(data, ranges=ranges, dtypes=dtypes):
ec.unique_index(data)
ec.none_missing(data)
ec.within_range(data, items=ranges)
ec.has_dtypes(data, items=dtypes)
def wrapper(*args, **kwargs):
result = func(*args, **kwargs)
ck.within_range(result, items)
return result
