def test_rename_columns():
"""Check that DataFrameSchema.rename_columns() method does its job"""
rename_dict = {"col1": "col1_new_name", "col2": "col2_new_name"}
schema_original = DataFrameSchema(columns={
"col1": Column(Int),
"col2": Column(Float)
})
schema_renamed = schema_original.rename_columns(rename_dict)
# Check if new column names are indeed present in the new schema
assert all([
col_name in rename_dict.values() for col_name in schema_renamed.columns
])
# Check if original schema didn't change in the process
assert all(
[col_name in schema_original.columns for col_name in rename_dict])
with pytest.raises(errors.SchemaInitError):
schema_original.rename_columns({"foo": "bar"})
def test_rename_columns():
"""Check that DataFrameSchema.rename_columns() method does it's job"""
rename_dict = {'col1': 'col1_new_name', 'col2': 'col2_new_name'}
schema_original = DataFrameSchema(columns={
'col1': Column(Int),
'col2': Column(Float)
})
schema_renamed = schema_original.rename_columns(rename_dict)
# Check if new column names are indeed present in the new schema
assert all([
col_name in rename_dict.values() for col_name in schema_renamed.columns
])
# Check if original schema didn't change in the process
assert all(
[col_name in schema_original.columns for col_name in rename_dict])
