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