Remove the duplicate `_with_context_manager` tests now that `Mesh` is the default way to create a mesh.
Remove the duplicate `_with_context_manager` tests now that `Mesh` is the default
way to create a mesh.

PiperOrigin-RevId: 436333536

google/jax,google/jax,google/jax,google/jax