def test_column_from_thrift(self): self.assertEqual( types.Column.from_thrift( ttypes.Column("A", ttypes.ColumnType(text_type=ttypes.ColumnTypeText())) ), types.Column("A", types.ColumnType.Text()), )
def test_column_type_text_from_thrift(self): self.assertEqual( types.thrift_column_type_to_arrow( ttypes.ColumnType(text_type=ttypes.ColumnTypeText()) ), types.ColumnType.Text(), )
def test_table_metadata_to_thrift(self): self.assertEqual( types.TableMetadata( 4, [ types.Column("A", types.ColumnType.Text()), types.Column("B", types.ColumnType.Text()), ], ).to_thrift(), ttypes.TableMetadata( 4, [ ttypes.Column( "A", ttypes.ColumnType(text_type=ttypes.ColumnTypeText()) ), ttypes.Column( "B", ttypes.ColumnType(text_type=ttypes.ColumnTypeText()) ), ], ), )
def test_column_type_text_to_thrift(self): self.assertEqual( types.ColumnType.Text().to_thrift(), ttypes.ColumnType(text_type=ttypes.ColumnTypeText()), )