def test_rename_slugify_columns_in_place(self):
column_names = ['Test kož', 'test 2', 'test 2']
table = Table(self.rows, column_names, self.column_types)
table2 = table.rename(slug_columns=True)
table3 = table.rename(slug_columns=True, separator='.')
self.assertColumnNames(table, ['Test kož', 'test 2', 'test 2_2'])
self.assertColumnNames(table2, ['test-koz', 'test-2', 'test-2-2'])
self.assertColumnNames(table3, ['test.koz', 'test.2', 'test.2.2'])
def test_rename_slugify_columns(self):
strings = ['Test kož', 'test 2', 'test 2']
table = Table(self.rows, self.column_names, self.column_types)
table2 = table.rename(strings, slug_columns=True)
table3 = table.rename(strings, slug_columns=True, separator='.')
self.assertColumnNames(table, ['one', 'two', 'three'])
self.assertColumnNames(table2, ['test_koz', 'test_2', 'test_2_2'])
self.assertColumnNames(table3, ['test.koz', 'test.2', 'test.2.2'])
def test_rename_slugify_columns(self):
strings = ['Test kož', 'test 2', 'test 2']
table = Table(self.rows, self.column_names, self.column_types)
table2 = table.rename(strings, slug_columns=True)
table3 = table.rename(strings, slug_columns=True, separator='.')
self.assertColumnNames(table, ['one', 'two', 'three'])
self.assertColumnNames(table2, ['test-koz', 'test-2', 'test-2-1'])
self.assertColumnNames(table3, ['test.koz', 'test.2', 'test.2.1'])
def test_rename_slugify_rows(self):
strings = ['Test kož', 'test 2', 'test 2']
table = Table(self.rows, self.column_names, self.column_types)
table2 = table.rename(row_names=strings, slug_rows=True)
table3 = table.rename(row_names=strings, slug_rows=True, separator='.')
self.assertIs(table.row_names, None)
self.assertRowNames(table2, ['test-koz', 'test-2', 'test-2-1'])
self.assertRowNames(table3, ['test.koz', 'test.2', 'test.2.1'])
def test_rename_slugify_columns_in_place(self):
column_names = ['Test kož', 'test 2', 'test 2']
table = Table(self.rows, column_names, self.column_types)
table2 = table.rename(slug_columns=True)
table3 = table.rename(slug_columns=True, separator='.')
self.assertColumnNames(table, ['Test kož', 'test 2', 'test 2_2'])
self.assertColumnNames(table2, ['test-koz', 'test-2', 'test-2-2'])
self.assertColumnNames(table3, ['test.koz', 'test.2', 'test.2.2'])
def test_rename_slugify_rows(self):
strings = ['Test kož', 'test 2', 'test 2']
table = Table(self.rows, self.column_names, self.column_types)
table2 = table.rename(row_names=strings, slug_rows=True)
table3 = table.rename(row_names=strings, slug_rows=True, separator='.')
self.assertIs(table.row_names, None)
self.assertRowNames(table2, ['test_koz', 'test_2', 'test_2_2'])
self.assertRowNames(table3, ['test.koz', 'test.2', 'test.2.2'])
def test_rename_slugify_rows_in_place(self):
strings = ['Test kož', 'test 2', 'test 2']
table = Table(self.rows,
self.column_names,
self.column_types,
row_names=strings)
table2 = table.rename(slug_rows=True)
table3 = table.rename(slug_rows=True, separator='.')
self.assertRowNames(table, ['Test kož', 'test 2', 'test 2'])
self.assertRowNames(table2, ['test-koz', 'test-2', 'test-2-1'])
self.assertRowNames(table3, ['test.koz', 'test.2', 'test.2.1'])
def _catalog_filter_table(cls, table: agate.Table,
manifest: Manifest) -> agate.Table:
# On snowflake, users can set QUOTED_IDENTIFIERS_IGNORE_CASE, so force
# the column names to their lowercased forms.
lowered = table.rename(
column_names=[c.lower() for c in table.column_names])
return super()._catalog_filter_table(lowered, manifest)
def _catalog_filter_table(cls, table: agate.Table,
manifest: Manifest) -> agate.Table:
table = table.rename(column_names={
col.name: col.name.replace('__', ':')
for col in table.columns
})
return super()._catalog_filter_table(table, manifest)
def test_rename_slugify_columns_in_place(self):
column_names = [u'Test kož', 'test 2', 'test 2']
warnings.simplefilter('ignore')
try:
table = Table(self.rows, column_names, self.column_types)
finally:
warnings.resetwarnings()
table2 = table.rename(slug_columns=True)
table3 = table.rename(slug_columns=True, separator='.')
self.assertColumnNames(table, [u'Test kož', 'test 2', 'test 2_2'])
self.assertColumnNames(table2, ['test_koz', 'test_2', 'test_2_2'])
self.assertColumnNames(table3, ['test.koz', 'test.2', 'test.2.2'])
def test_rename_column_names_renames_row_values(self):
table = Table(self.rows, self.column_names, self.column_types)
new_column_names = ['d', 'e', 'f']
table2 = table.rename(column_names=new_column_names)
self.assertColumnNames(table2, new_column_names)
def test_rename_column_names_renames_row_values(self):
table = Table(self.rows, self.column_names, self.column_types)
new_column_names = ['d', 'e', 'f']
table2 = table.rename(column_names=new_column_names)
self.assertColumnNames(table2, new_column_names)
def _catalog_filter_table(
cls, table: agate.Table, manifest: Manifest
) -> agate.Table:
lowered = table.rename(
column_names=[c.lower() for c in table.column_names]
)
return super()._catalog_filter_table(lowered, manifest)
def test_rename_column_names_dict(self):
table = Table(self.rows, self.column_names, self.column_types)
table2 = table.rename(column_names={'two': 'second'})
self.assertIs(table2.row_names, None)
self.assertSequenceEqual(table2.column_names, ['one', 'second', 'three'])
self.assertIs(table.row_names, None)
self.assertSequenceEqual(table.column_names, self.column_names)
def test_rename_column_names(self):
table = Table(self.rows, self.column_names, self.column_types)
table2 = table.rename(column_names=['d', 'e', 'f'])
self.assertIs(table2.row_names, None)
self.assertSequenceEqual(table2.column_names, ['d', 'e', 'f'])
self.assertIs(table.row_names, None)
self.assertSequenceEqual(table.column_names, self.column_names)
def test_rename_row_names_dict(self):
table = Table(self.rows, self.column_names, self.column_types, row_names=['a', 'b', 'c'])
table2 = table.rename(row_names={'b': 'd'})
self.assertSequenceEqual(table2.row_names, ['a', 'd', 'c'])
self.assertSequenceEqual(table2.column_names, self.column_names)
self.assertSequenceEqual(table.row_names, ['a', 'b', 'c'])
self.assertSequenceEqual(table.column_names, self.column_names)
def test_rename_column_names_dict(self):
table = Table(self.rows, self.column_names, self.column_types)
table2 = table.rename(column_names={'two': 'second'})
self.assertIs(table2.row_names, None)
self.assertSequenceEqual(table2.column_names, ['one', 'second', 'three'])
self.assertIs(table.row_names, None)
self.assertSequenceEqual(table.column_names, self.column_names)
def test_rename_column_names(self):
table = Table(self.rows, self.column_names, self.column_types)
table2 = table.rename(column_names=['d', 'e', 'f'])
self.assertIs(table2.row_names, None)
self.assertSequenceEqual(table2.column_names, ['d', 'e', 'f'])
self.assertIs(table.row_names, None)
self.assertSequenceEqual(table.column_names, self.column_names)
def test_rename_row_names_dict(self):
table = Table(self.rows, self.column_names, self.column_types, row_names=['a', 'b', 'c'])
table2 = table.rename(row_names={'b': 'd'})
self.assertSequenceEqual(table2.row_names, ['a', 'd', 'c'])
self.assertSequenceEqual(table2.column_names, self.column_names)
self.assertSequenceEqual(table.row_names, ['a', 'b', 'c'])
self.assertSequenceEqual(table.column_names, self.column_names)
