def METHOD_NAME(interpolation_method):
    """Test that interpolation method works as expected."""
    latent1 = torch.randn(100, 25)
    latent2 = torch.randn(100, 25)
    res1 = _interpolate(latent1, latent2, 1e-4, interpolation_method)
    res2 = batch_interp(latent1, latent2, 1e-4, interpolation_method)
    assert torch.allclose(res1, res2)