def METHOD_NAME(use_torch, msbg_model):
    """Test the measure_rms function"""
    x = torch.randn(2, 20000)
    x = x.cpu()
    y_torch = msbg_model.measure_rms(x)
    y = y_torch.cpu().detach().numpy()
    assert y.shape == (2, 1)
    assert y[0, 0] == pytest.approx(
        1.0040439, rel=pytest.rel_tolerance, abs=pytest.abs_tolerance
    )
    assert y[1, 0] == pytest.approx(
        0.99920964, rel=pytest.rel_tolerance, abs=pytest.abs_tolerance
    )