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)
