def test_df_equality(self):
df_col = DataFrameColumn('name', ColumnType.TEXT, is_nullable=False)
self.assertEqual(df_col, df_col)
df_col1 = DataFrameColumn('name2', ColumnType.TEXT, is_nullable=False)
self.assertNotEqual(df_col, df_col1)
df_col1 = DataFrameColumn('name',
ColumnType.INTEGER,
is_nullable=False)
self.assertNotEqual(df_col, df_col1)
df_col1 = DataFrameColumn('name', ColumnType.INTEGER, is_nullable=True)
self.assertNotEqual(df_col, df_col1)
df_col1 = DataFrameColumn('name',
ColumnType.INTEGER,
is_nullable=False)
self.assertNotEqual(df_col, df_col1)
df_col.array_dimensions = [2, 4]
df_col1 = DataFrameColumn('name',
ColumnType.INTEGER,
is_nullable=False,
array_dimensions=[1, 2])
self.assertNotEqual(df_col, df_col1)
df_col.metadata_id = 1
df_col1 = DataFrameColumn('name',
ColumnType.INTEGER,
is_nullable=False,
array_dimensions=[2, 4],
metadata_id=2)
self.assertNotEqual(df_col, df_col1)
def test_df_column(self):
df_col = DataFrameColumn('name', ColumnType.TEXT, is_nullable=False)
df_col.array_dimensions = [1, 2]
df_col.metadata_id = 1
self.assertEqual(df_col.array_dimensions, [1, 2])
self.assertEqual(df_col.is_nullable, False)
self.assertEqual(df_col.name, 'name')
self.assertEqual(df_col.type, ColumnType.TEXT)
self.assertEqual(df_col.metadata_id, 1)
self.assertEqual(df_col.id, None)
self.assertEqual(str(df_col), 'Column: (name, TEXT, False, [1, 2])')