test: skip tests of deprecated stuff

Author: beckermr

tests/test_deprecated.py

@@ -35,12 +35,10 @@ def check_dep(f, *args, **kwargs):
 
 @timer
 def test_gsparams():
-    if is_jax_galsim():
-        pass
-    else:
-        check_dep(galsim.GSParams, allowed_flux_variation=0.90)
-        check_dep(galsim.GSParams, range_division_for_extrema=50)
-        check_dep(galsim.GSParams, small_fraction_of_flux=1.e-6)
+    if is_jax_galsim(): return
+    check_dep(galsim.GSParams, allowed_flux_variation=0.90)
+    check_dep(galsim.GSParams, range_division_for_extrema=50)
+    check_dep(galsim.GSParams, small_fraction_of_flux=1.e-6)
 
 
 @timer
@@ -498,6 +496,8 @@ def test_hsm_depr():
 
 @timer
 def test_photon_array_depr():
+    if is_jax_galsim(): return
+
     nphotons = 1000
 
     # First create from scratch
@@ -544,203 +544,96 @@ def test_photon_array_depr():
 
     # Using the getter is allowed, but deprecated.
     photon_array = galsim.PhotonArray(nphotons)
-    if is_jax_galsim():
-        # jax-galsim always sets these additional properties
-        # dxdz = check_dep(getattr, photon_array, 'dxdz')
-        # however jax-galsim sets them to NaN so they are not allocated
-        assert not photon_array.hasAllocatedAngles()
-        assert len(photon_array.dxdz) == nphotons
-        # JAX-Galsim does not allow by reference setting - changed this
-        # to make tests below run
-        photon_array.dxdz = 0.17
-        # non-nan means allocated for jax-galsim
-        assert photon_array.hasAllocatedAngles()
-    else:
-        dxdz = check_dep(getattr, photon_array, 'dxdz')
-        assert photon_array.hasAllocatedAngles()
-        assert photon_array.hasAllocatedAngles()
-        assert len(photon_array.dxdz) == nphotons
-        assert len(photon_array.dydz) == nphotons
-        dxdz[:] = 0.17
+    dxdz = check_dep(getattr, photon_array, 'dxdz')
+    assert photon_array.hasAllocatedAngles()
+    assert photon_array.hasAllocatedAngles()
+    assert len(photon_array.dxdz) == nphotons
+    assert len(photon_array.dydz) == nphotons
+    dxdz[:] = 0.17
     np.testing.assert_array_equal(photon_array.dxdz, 0.17)
     np.testing.assert_array_equal(photon_array.dydz, 0.)
 
-    if is_jax_galsim():
-        assert hasattr(photon_array, "dydz")
-        # JAX-Galsim does not allow by reference setting - changed this
-        # to make tests below run
-        photon_array.dydz = 0.59
-    else:
-        dydz = photon_array.dydz  # Allowed now.
-        dydz[:] = 0.59
+    dydz = photon_array.dydz  # Allowed now.
+    dydz[:] = 0.59
     np.testing.assert_array_equal(photon_array.dydz, 0.59)
 
-    if is_jax_galsim():
-        # jax-galsim always sets these additional properties
-        # wave = check_dep(getattr, photon_array, 'wavelength')
-        # however jax-galsim sets them to NaN so they are not allocated
-        assert not photon_array.hasAllocatedWavelengths()
-    else:
-        wave = check_dep(getattr, photon_array, 'wavelength')
-        assert photon_array.hasAllocatedWavelengths()
+    wave = check_dep(getattr, photon_array, 'wavelength')
+    assert photon_array.hasAllocatedWavelengths()
     assert len(photon_array.wavelength) == nphotons
-    if is_jax_galsim():
-        # JAX-Galsim does not allow by reference setting - changed this
-        # to make tests below run
-        photon_array.wavelength = 500.0
-        # jax-galsim is allocated now
-        assert photon_array.hasAllocatedWavelengths()
-    else:
-        wave[:] = 500.
+    wave[:] = 500.
     np.testing.assert_array_equal(photon_array.wavelength, 500)
 
-    if is_jax_galsim():
-        # jax-galsim always sets these additional properties
-        # u = check_dep(getattr, photon_array, "pupil_u")
-        # however jax-galsim sets them to NaN so they are not allocated
-        assert not photon_array.hasAllocatedPupil()
-    else:
-        u = check_dep(getattr, photon_array, 'pupil_u')
-        assert photon_array.hasAllocatedPupil()
-        assert len(photon_array.pupil_u) == nphotons
+    u = check_dep(getattr, photon_array, 'pupil_u')
+    assert photon_array.hasAllocatedPupil()
+    assert len(photon_array.pupil_u) == nphotons
     assert len(photon_array.pupil_v) == nphotons
-    if is_jax_galsim():
-        # JAX-Galsim does not allow by reference setting - changed this
-        # to make tests below run
-        photon_array.pupil_u = 6.0
-        # jax-galsim is allocated now
-        assert photon_array.hasAllocatedPupil()
-    else:
-        u[:] = 6.0
+    u[:] = 6.0
     np.testing.assert_array_equal(photon_array.pupil_u, 6.0)
     np.testing.assert_array_equal(photon_array.pupil_v, 0.0)
-    if is_jax_galsim():
-        assert hasattr(photon_array, "pupil_v")
-        # JAX-Galsim does not allow by reference setting - changed this
-        # to make tests below run
-        photon_array.pupil_v = 10.0
-    else:
-        v = photon_array.pupil_v
-        v[:] = 10.0
+    v = photon_array.pupil_v
+    v[:] = 10.0
     np.testing.assert_array_equal(photon_array.pupil_v, 10.0)
 
-    if is_jax_galsim():
-        # jax-galsim always sets these additional properties
-        # t = check_dep(getattr, photon_array, "time")
-        # however jax-galsim sets them to NaN so they are not allocated
-        assert not photon_array.hasAllocatedTimes()
-        # jax-galsim needs to set 0
-        photon_array.time = 0.0
-        # jax-galsim is allocated now
-        assert photon_array.hasAllocatedTimes()
-    else:
-        t = check_dep(getattr, photon_array, 'time')
+    t = check_dep(getattr, photon_array, 'time')
     assert photon_array.hasAllocatedTimes()
     assert len(photon_array.time) == nphotons
     np.testing.assert_array_equal(photon_array.time, 0.0)
-    if is_jax_galsim():
-        # JAX-Galsim does not allow by reference setting - changed this
-        # to make tests below run
-        photon_array.time = 10
-    else:
-        t[:] = 10
+    t[:] = 10
     np.testing.assert_array_equal(photon_array.time, 10.0)
 
     # For coverage, also need to test the two pair ones in other order.
     photon_array = galsim.PhotonArray(nphotons)
-    if is_jax_galsim():
-        # jax-galsim always sets these additional properties
-        # dydz = check_dep(getattr, photon_array, "dydz")
-        # however jax-galsim sets them to NaN so they are not allocated
-        assert not photon_array.hasAllocatedAngles()
-    else:
-        dydz = check_dep(getattr, photon_array, 'dydz')
-        assert photon_array.hasAllocatedAngles()
-        assert photon_array.hasAllocatedAngles()
+    dydz = check_dep(getattr, photon_array, 'dydz')
+    assert photon_array.hasAllocatedAngles()
+    assert photon_array.hasAllocatedAngles()
     assert len(photon_array.dxdz) == nphotons
     assert len(photon_array.dydz) == nphotons
-    if is_jax_galsim():
-        # JAX-Galsim does not allow by reference setting - changed this
-        # to make tests below run
-        photon_array.dydz = 0.59
-        # non-nan means allocated for jax-galsim
-        assert photon_array.hasAllocatedAngles()
-    else:
-        dydz[:] = 0.59
+    dydz[:] = 0.59
     np.testing.assert_array_equal(photon_array.dxdz, 0.)
     np.testing.assert_array_equal(photon_array.dydz, 0.59)
 
-    if is_jax_galsim():
-        assert hasattr(photon_array, "dxdz")
-        # JAX-Galsim does not allow by reference setting - changed this
-        # to make tests below run
-        photon_array.dxdz = 0.17
-    else:
-        dxdz = photon_array.dxdz  # Allowed now.
-        dxdz[:] = 0.17
+    dxdz = photon_array.dxdz  # Allowed now.
+    dxdz[:] = 0.17
     np.testing.assert_array_equal(photon_array.dxdz, 0.17)
 
-    if is_jax_galsim():
-        # jax-galsim always sets these additional properties
-        # v = check_dep(getattr, photon_array, "pupil_v")
-        # however jax-galsim sets them to NaN so they are not allocated
-        assert not photon_array.hasAllocatedPupil()
-    else:
-        v = check_dep(getattr, photon_array, 'pupil_v')
-        assert photon_array.hasAllocatedPupil()
-        assert len(photon_array.pupil_u) == nphotons
+    v = check_dep(getattr, photon_array, 'pupil_v')
+    assert photon_array.hasAllocatedPupil()
+    assert len(photon_array.pupil_u) == nphotons
     assert len(photon_array.pupil_v) == nphotons
-    if is_jax_galsim():
-        # JAX-Galsim does not allow by reference setting - changed this
-        # to make tests below run
-        photon_array.pupil_v = 10.0
-        # jax-galsim is allocated now
-        assert photon_array.hasAllocatedPupil()
-    else:
-        v[:] = 10.0
+    v[:] = 10.0
     np.testing.assert_array_equal(photon_array.pupil_u, 0.0)
     np.testing.assert_array_equal(photon_array.pupil_v, 10.0)
-    if is_jax_galsim():
-        assert hasattr(photon_array, "pupil_u")
-        # JAX-Galsim does not allow by reference setting - changed this
-        # to make tests below run
-        photon_array.pupil_u = 6.0
-    else:
-        u = photon_array.pupil_u
-        u[:] = 6.0
+    u = photon_array.pupil_u
+    u[:] = 6.0
     np.testing.assert_array_equal(photon_array.pupil_u, 6.0)
 
     # Check assignAt
     pa1 = galsim.PhotonArray(50)
     pa1.x = photon_array.x[:50]
-    if is_jax_galsim():
-        pa1.y = photon_array.y[:50]
-        pa1.flux.at[0:50].set(photon_array.flux[:50])
-    else:
-        for i in range(50):
-            pa1.y[i] = photon_array.y[i]
-        pa1.flux[0:50] = photon_array.flux[:50]
+    for i in range(50):
+        pa1.y[i] = photon_array.y[i]
+    pa1.flux[0:50] = photon_array.flux[:50]
     pa1.dxdz = photon_array.dxdz[:50]
     pa1.dydz = photon_array.dydz[:50]
     pa1.pupil_u = photon_array.pupil_u[:50]
     pa1.pupil_v = photon_array.pupil_v[:50]
     pa2 = galsim.PhotonArray(100)
     check_dep(pa2.assignAt, 0, pa1)
     check_dep(pa2.assignAt, 50, pa1)
-    np.testing.assert_array_almost_equal(pa2.x[:50], pa1.x)
-    np.testing.assert_array_almost_equal(pa2.y[:50], pa1.y)
-    np.testing.assert_array_almost_equal(pa2.flux[:50], pa1.flux)
-    np.testing.assert_array_almost_equal(pa2.dxdz[:50], pa1.dxdz)
-    np.testing.assert_array_almost_equal(pa2.dydz[:50], pa1.dydz)
-    np.testing.assert_array_almost_equal(pa2.pupil_u[:50], pa1.pupil_u)
-    np.testing.assert_array_almost_equal(pa2.pupil_v[:50], pa1.pupil_v)
-    np.testing.assert_array_almost_equal(pa2.x[50:], pa1.x)
-    np.testing.assert_array_almost_equal(pa2.y[50:], pa1.y)
-    np.testing.assert_array_almost_equal(pa2.flux[50:], pa1.flux)
-    np.testing.assert_array_almost_equal(pa2.dxdz[50:], pa1.dxdz)
-    np.testing.assert_array_almost_equal(pa2.dydz[50:], pa1.dydz)
-    np.testing.assert_array_almost_equal(pa2.pupil_u[50:], pa1.pupil_u)
-    np.testing.assert_array_almost_equal(pa2.pupil_v[50:], pa1.pupil_v)
+    np.testing.assert_almost_equal(pa2.x[:50], pa1.x)
+    np.testing.assert_almost_equal(pa2.y[:50], pa1.y)
+    np.testing.assert_almost_equal(pa2.flux[:50], pa1.flux)
+    np.testing.assert_almost_equal(pa2.dxdz[:50], pa1.dxdz)
+    np.testing.assert_almost_equal(pa2.dydz[:50], pa1.dydz)
+    np.testing.assert_almost_equal(pa2.pupil_u[:50], pa1.pupil_u)
+    np.testing.assert_almost_equal(pa2.pupil_v[:50], pa1.pupil_v)
+    np.testing.assert_almost_equal(pa2.x[50:], pa1.x)
+    np.testing.assert_almost_equal(pa2.y[50:], pa1.y)
+    np.testing.assert_almost_equal(pa2.flux[50:], pa1.flux)
+    np.testing.assert_almost_equal(pa2.dxdz[50:], pa1.dxdz)
+    np.testing.assert_almost_equal(pa2.dydz[50:], pa1.dydz)
+    np.testing.assert_almost_equal(pa2.pupil_u[50:], pa1.pupil_u)
+    np.testing.assert_almost_equal(pa2.pupil_v[50:], pa1.pupil_v)
 
     # Error if it doesn't fit.
     with assert_raises(ValueError):