diff --git a/tests/test_exponential.py b/tests/test_exponential.py index 68900e40f1..52569c4131 100644 --- a/tests/test_exponential.py +++ b/tests/test_exponential.py @@ -231,12 +231,7 @@ def test_exponential_shoot(): assert np.isclose(added_flux, obj.flux) assert np.isclose(im.array.sum(), obj.flux) photons2 = obj.makePhot(poisson_flux=False, rng=rng) - if is_jax_galsim(): - np.testing.assert_allclose(photons2.x, photons.x) - np.testing.assert_allclose(photons2.y, photons.y) - np.testing.assert_allclose(photons2.flux, photons.flux) - else: - assert photons2 == photons, "Exponential makePhot not equivalent to drawPhot" + assert photons2 == photons, "Exponential makePhot not equivalent to drawPhot" @timer