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_params_to_thrift(self):
self.assertEqual(
types.Params({
"str":
"s",
"int":
2,
"float":
1.2,
"null":
None,
"bool":
False,
"column":
types.Column("A", types.ColumnType.Number(format="{:,.2f}")),
"listofmaps": [{
"A": "a",
"B": "b"
}, {
"C": "c",
"D": "d"
}],
"tab":
"TODO tabs",
}).to_thrift(),
{
"str":
ttypes.ParamValue(string_value="s"),
"int":
ttypes.ParamValue(integer_value=2),
"float":
ttypes.ParamValue(float_value=1.2),
"null":
ttypes.ParamValue(),
"bool":
ttypes.ParamValue(boolean_value=False),
"column":
ttypes.ParamValue(column_value=ttypes.Column(
"A",
ttypes.ColumnType(number_type=ttypes.ColumnTypeNumber(
format="{:,.2f}")),
)),
"listofmaps":
ttypes.ParamValue(list_value=[
ttypes.ParamValue(
map_value={
"A": ttypes.ParamValue(string_value="a"),
"B": ttypes.ParamValue(string_value="b"),
}),
ttypes.ParamValue(
map_value={
"C": ttypes.ParamValue(string_value="c"),
"D": ttypes.ParamValue(string_value="d"),
}),
]),
"tab":
ttypes.ParamValue(string_value="TODO tabs"),
},
)
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())
),
],
),
)
