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()