def test_handle(model_hdlr: ModelHandler, rqst: Request, context: Context, num_samples: int, quantiles: List[str]): model_hdlr.initialize(context) results_list = model_hdlr.handle([rqst], context) # Make sure each JSON line is somewhat correct. # for each request for results_list_request in results_list: for line in io.StringIO(results_list_request): d = json.loads(line) for quantile in quantiles: assert quantile in d["quantiles"] assert len(d["quantiles"][quantile]) == model_hdlr.mx_model.prediction_length assert ("mean" in d) and (len(d["mean"]) == model_hdlr.mx_model.prediction_length) # Print results; need to run pytest -v -rA --tb=short ... print(results_list) return results_list
def test_initialize(model_hdlr: ModelHandler, context: Context): model_hdlr.initialize(context) predictor = model_hdlr.mx_model assert isinstance(predictor, Predictor)