def test_combine_dfs_2(): """Check combine_dfs operator when the dataframes are equal""" tpot_obj = TPOT() df1 = pd.DataFrame({'a': range(10), 'b': range(10, 20)}) df2 = pd.DataFrame({'a': range(10), 'b': range(10, 20)}) combined_df = pd.DataFrame({'a': range(10), 'b': range(10, 20)}) assert tpot_obj._combine_dfs(df1, df2).equals(combined_df)
def test_combine_dfs(): tpot_obj = TPOT() df1 = pd.DataFrame({'a': range(10), 'b': range(10, 20)}) df2 = pd.DataFrame({'b': range(10, 20), 'c': range(20, 30)}) combined_df = pd.DataFrame({'a': range(10), 'b': range(10, 20), 'c': range(20, 30)}) assert tpot_obj._combine_dfs(df1, df2).equals(combined_df)
def test_combine_dfs(): tpot_obj = TPOT() df1 = pd.DataFrame({'a': range(10), 'b': range(10, 20)}) df2 = pd.DataFrame({'b': range(10, 20), 'c': range(20, 30)}) combined_df = pd.DataFrame({ 'a': range(10), 'b': range(10, 20), 'c': range(20, 30) }) assert tpot_obj._combine_dfs(df1, df2).equals(combined_df)