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_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)
