test: just do less for jax

GalSim-developers · Sep 13, 2024 · fb890ba · fb890ba
1 parent 501c0fd
commit fb890ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_interpolatedimage.py b/tests/test_interpolatedimage.py
@@ -424,7 +424,7 @@ def test_unit_integrals():
         print('integrals: ',len(integrals),integrals)
 
         assert len(integrals) == n
-        np.testing.assert_allclose(integrals[_n_do], direct_integrals[_n_do], atol=1.e-12)
+        np.testing.assert_allclose(integrals[:_n_do], direct_integrals[:_n_do], atol=1.e-12)
 
         if n > 10:
             print('n>10 for ',repr(interp))