improve test

Author: ferrine

tests/scalar/test_loop.py

@@ -213,10 +213,16 @@ def test_inner_composite(mode):
     assert y16.type.dtype == "float16"
 
     fn16 = function([n_steps, x16], y16, mode=mode)
-    out16 = fn16(n_steps=9, x16=np.array(4.73, dtype="float16"))
-    assert out16.dtype == "float16"
+    out16 = fn16(n_steps=3, x16=np.array(4.73, dtype="float16"))
+    np.testing.assert_allclose(
+        out16,
+        4.73 + 3,
+        rtol=1e-3,
+    )
+    out16overflow = fn16(n_steps=9, x16=np.array(4.73, dtype="float16"))
+    assert out16overflow.dtype == "float16"
     # with this dtype overflow happens
-    assert np.isnan(out16)
+    assert np.isnan(out16overflow)
 
 
 @mode