def test_unique():
df = pd.DataFrame([[1, 2, 3], ['a', 'b', 'c']])
tm.assert_frame_equal(df, ck.unique(df))
result = dc.unique()(_noop)(df)
tm.assert_frame_equal(result, df)
df = pd.DataFrame([[1, 2, 3], [1, 'b', 'c']])
with pytest.raises(AssertionError):
ck.unique(df)
with pytest.raises(AssertionError):
dc.unique()(_noop)(df)
def wrapper(*args, **kwargs):
result = func(*args, **kwargs)
ck.unique(result, columns=columns)
return result
def wrapper(*args, **kwargs):
result = func(*args, **kwargs)
ck.unique(result, columns=columns)
return result