def METHOD_NAME(self):
    # Create model
    model = pybamm.BaseModel()
    model.convert_to_format = "jax"
    domain = ["negative electrode", "separator", "positive electrode"]
    var = pybamm.Variable("var", domain=domain)
    model.rhs = {var: -pybamm.InputParameter("rate") * var}
    model.initial_conditions = {var: 1}
    # create discretisation
    mesh = get_mesh_for_testing()
    spatial_methods = {"macroscale": pybamm.FiniteVolume()}
    disc = pybamm.Discretisation(mesh, spatial_methods)
    disc.process_model(model)
    # Solve
    t_eval = np.linspace(0, 10, 80)
    y0 = model.concatenated_initial_conditions.evaluate().reshape(-1)
    rhs = pybamm.EvaluatorJax(model.concatenated_rhs)
    def fun(y, t, inputs):
        return rhs(t=t, y=y, inputs=inputs).reshape(-1)
    y = pybamm.jax_bdf_integrate(
        fun, y0, t_eval, {"rate": 0.1}, rtol=1e-9, atol=1e-9
    )
    np.testing.assert_allclose(y[:, 0].reshape(-1), np.exp(-0.1 * t_eval))