test: enable tests for errors for image

Author: beckermr

tests/test_draw.py

@@ -380,24 +380,18 @@ def test_drawImage():
         os.path.join(os.path.dirname(__file__), 'fits_files/tpv.fits')))
 
     assert_raises(ValueError, obj.drawImage, bounds=galsim.BoundsI())
-    if is_jax_galsim():
-        pass
-    else:
-        assert_raises(ValueError, obj.drawImage, image=im10, gain=0.)
-        assert_raises(ValueError, obj.drawImage, image=im10, gain=-1.)
-        assert_raises(ValueError, obj.drawImage, image=im10, area=0.)
-        assert_raises(ValueError, obj.drawImage, image=im10, area=-1.)
-        assert_raises(ValueError, obj.drawImage, image=im10, exptime=0.)
-        assert_raises(ValueError, obj.drawImage, image=im10, exptime=-1.)
+    assert_raises((ValueError, Exception), obj.drawImage, image=im10, gain=0.)
+    assert_raises((ValueError, Exception), obj.drawImage, image=im10, gain=-1.)
+    assert_raises((ValueError, Exception), obj.drawImage, image=im10, area=0.)
+    assert_raises((ValueError, Exception), obj.drawImage, image=im10, area=-1.)
+    assert_raises((ValueError, Exception), obj.drawImage, image=im10, exptime=0.)
+    assert_raises((ValueError, Exception), obj.drawImage, image=im10, exptime=-1.)
     assert_raises(ValueError, obj.drawImage, image=im10, method='invalid')
 
     # These options are invalid unless metho=phot
     assert_raises(TypeError, obj.drawImage, image=im10, n_photons=3)
     assert_raises(TypeError, obj.drawImage, rng=galsim.BaseDeviate(234))
-    if is_jax_galsim():
-        pass
-    else:
-        assert_raises(TypeError, obj.drawImage, max_extra_noise=23)
+    assert_raises(ValueError, obj.drawImage, max_extra_noise=23)
     assert_raises(TypeError, obj.drawImage, poisson_flux=True)
     assert_raises(TypeError, obj.drawImage, maxN=10000)
     assert_raises(TypeError, obj.drawImage, save_photons=True)