TST: polyval NEP 50 return type test

numpy · May 24, 2024 · 8e431a6 · 8e431a6
1 parent 39fc96a
commit 8e431a6
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/numpy/lib/tests/test_polynomial.py b/numpy/lib/tests/test_polynomial.py
@@ -301,3 +301,13 @@ def test_poly_coeffs_mutable(self):
         # this never used to be allowed - let's not add features to deprecated
         # APIs
         assert_raises(AttributeError, setattr, p, 'coeffs', np.array(1))
+
+    def test_polyval_return_types(self):
+        v = np.polyval(np.array([5.], dtype=np.float32), 5.0)
+        assert_equal(v.dtype, np.float32)
+
+        v = np.polyval(np.array([5.], dtype=np.float16), 5.0)
+        assert_equal(v.dtype, np.float16)
+
+        v = np.polyval([5.], np.float16(5.0))
+        assert_equal(v.dtype, np.float64)