def test_unique_index():
df = pd.DataFrame([1, 2, 3], index=['a', 'b', 'c'])
tm.assert_frame_equal(df, ck.unique_index(df))
result = dc.unique_index()(_add_n)(df)
tm.assert_frame_equal(result, df + 1)
with pytest.raises(AssertionError):
ck.unique_index(df.reindex(['a', 'a', 'b']))
with pytest.raises(AssertionError):
dc.unique_index()(_add_n)(df.reindex(['a', 'a', 'b']))
def test_unique_index():
df = pd.DataFrame([1, 2, 3], index=['a', 'b', 'c'])
tm.assert_frame_equal(df, ck.unique_index(df))
result = dc.unique_index()(_add_n)(df)
tm.assert_frame_equal(result, df + 1)
with pytest.raises(AssertionError):
ck.unique_index(df.reindex(['a', 'a', 'b']))
with pytest.raises(AssertionError):
dc.unique_index()(_add_n)(df.reindex(['a', 'a', 'b']))
def wrapper(*args, **kwargs):
result = func(*args, **kwargs)
ck.unique_index(result)
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.unique_index(result)
return result
