def test_raw_params_to_thrift(self):
self.assertEqual(
types.RawParams({
"A": "x",
"B": [1, 2]
}).to_thrift(),
ttypes.RawParams('{"A":"x","B":[1,2]}'),
)
def test_raw_params_from_thrift(self):
self.assertEqual(
types.thrift_raw_params_to_arrow(
ttypes.RawParams('{"A":"x","B":[1,2]}')),
types.RawParams({
"A": "x",
"B": [1, 2]
}),
)
def migrate_params_thrift(params: ttypes.RawParams):
params_dict: Dict[str, Any] = thrift_raw_params_to_arrow(params).params
result_dict = migrate_params(params_dict)
return arrow_raw_params_to_thrift(types.RawParams(result_dict))
def migrate_params_thrift(params: ttypes.RawParams):
params_dict: Dict[str, Any] = types.RawParams.from_thrift(params).params
result_dict = migrate_params(params_dict)
return types.RawParams(result_dict).to_thrift()
