def METHOD_NAME(self):
    with self.model:
        idata = to_inference_data(self.trace[self.burn :])
    rhat = az.rhat(idata)
    for var in rhat:
        npt.assert_allclose(rhat[var], 1, rtol=0.01)