def create_sample_table(self):
table_info = DataFrameMetadata("dataset", 'dataset')
column_1 = DataFrameColumn("id", ColumnType.INTEGER, False)
column_2 = DataFrameColumn("data", ColumnType.NDARRAY, False,
[2, 2, 3])
table_info.schema = [column_1, column_2]
return table_info
def test_df_metadata_equality(self):
df_metadata = DataFrameMetadata('name', 'eva_dataset')
column_1 = DataFrameColumn("frame_id", ColumnType.INTEGER, False)
column_2 = DataFrameColumn("frame_label", ColumnType.INTEGER, False)
col_list = [column_1, column_2]
df_metadata.schema = col_list
self.assertEqual(df_metadata, df_metadata)
df_metadata1 = DataFrameMetadata('name2', 'eva_dataset')
column_1 = DataFrameColumn("frame_id", ColumnType.INTEGER, False)
column_2 = DataFrameColumn("frame_label", ColumnType.INTEGER, False)
col_list = [column_1, column_2]
df_metadata1.schema = col_list
self.assertNotEqual(df_metadata, df_metadata1)
df_metadata2 = DataFrameMetadata('name2', 'eva_dataset')
df_metadata2.schema = col_list[1:]
self.assertNotEqual(df_metadata1, df_metadata2)
def test_df_metadata(self):
df_metadata = DataFrameMetadata('name', 'eva_dataset')
column_1 = DataFrameColumn("frame_id", ColumnType.INTEGER, False)
column_2 = DataFrameColumn("frame_label", ColumnType.INTEGER, False)
col_list = [column_1, column_2]
schema = DataFrameSchema('name', col_list)
df_metadata.schema = col_list
self.assertEqual(df_metadata.name, 'name')
self.assertEqual(df_metadata.file_url, 'eva_dataset')
self.assertEqual(df_metadata.id, None)
self.assertEqual(df_metadata.identifier_column, 'id')
self.assertEqual(df_metadata.schema, schema)