def test_step(x, model, estimator, **config):
    """Test the `model` given the mini-batch of observations `x` and the gradient estimator/evaluator `estimator`"""
    loss, diagnostics, output = estimator(model, x, backward=False, **config)
    return diagnostics