def test_within_n_std(): df = pd.DataFrame({'A': np.arange(10), 'B': list('abcde') * 2}) tm.assert_frame_equal(df, ck.within_n_std(df)) tm.assert_frame_equal(df, dc.within_n_std()(_noop)(df)) with pytest.raises(AssertionError): ck.within_n_std(df, .5) with pytest.raises(AssertionError): dc.within_n_std(.5)(_noop)(df)
def test_within_n_std(): df = pd.DataFrame({'A': np.arange(10)}) tm.assert_frame_equal(df, ck.within_n_std(df)) tm.assert_frame_equal(df, dc.within_n_std()(_noop)(df)) with pytest.raises(AssertionError): ck.within_n_std(df, .5) with pytest.raises(AssertionError): dc.within_n_std(.5)(_noop)(df)
def wrapper(*args, **kwargs): result = func(*args, **kwargs) ck.within_n_std(result, n=n) return result