def test_is_same_as(): df = pd.DataFrame({'A': [1, 2, 3], 'B': [1, 2, 3]}) df_equal = pd.DataFrame({'A': [1, 2, 3], 'B': [1, 2, 3]}) df_not_equal = pd.DataFrame({'A': [1, 2, 3], 'B': [1, 2, 1]}) result = ck.is_same_as(df, df_equal) tm.assert_frame_equal(df, result) result = dc.is_same_as(df_equal)(_noop)(df) tm.assert_frame_equal(df, result) with pytest.raises(AssertionError): ck.is_same_as(df, df_not_equal) dc.is_same_as(df_not_equal)(_noop)(df)
def test_is_same_as_with_kwargs(): df = pd.DataFrame({'A': [1, 2, 3], 'B': [1, 2, 3]}) df_equal = pd.DataFrame({'A': [1, 2, 3], 'B': [1, 2, 3]}) df_equal_float = pd.DataFrame({'A': [1.0, 2, 3], 'B': [1, 2, 3.0]}) result = ck.is_same_as(df, df_equal, check_dtype=True) tm.assert_frame_equal(df, result) result = dc.is_same_as(df_equal, check_dtype=True)(_noop)(df) tm.assert_frame_equal(df, result) with pytest.raises(AssertionError): ck.is_same_as(df, df_equal_float, check_dtype=True) dc.is_same_as(df_equal_float, check_dtype=True)(_noop)(df) result = ck.is_same_as(df, df_equal_float, check_dtype=False) tm.assert_frame_equal(df, result) result = dc.is_same_as(df_equal_float, check_dtype=False)(_noop)(df) tm.assert_frame_equal(df, result)
def wrapper(*args, **kwargs): result = func(*args, **kwargs) ck.is_same_as(result, df_to_compare, **assert_kwargs) return result
def wrapper(*args, **kwargs): result = func(*args, **kwargs) ck.is_same_as(result, df_to_compare, **assert_kwargs) return result