Add a test I forgot to put in

GalSim-developers · Nov 1, 2023 · a93615f · a93615f
1 parent 37cc6a6
commit a93615f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_sensor.py b/tests/test_sensor.py
@@ -893,6 +893,8 @@ def test_treerings():
     im.setCenter(2000,2000)
     obj.drawImage(im, method='phot', sensor=sensor6, rng=rng6)
     print('im.sum = ',im.array.sum(), im.added_flux)
+    np.testing.assert_allclose(im.array.sum(), im.added_flux)
+    np.testing.assert_allclose(im.array.sum(), init_flux, rtol=1.e-3)
 
     areas6a = sensor6.calculate_pixel_areas(im, use_flux=True)
     assert np.min(areas6a.array) > 0