BUG: fix asarray_chkfinite to take dtype and order args, as advertised

Includes a doctest for dtype.

1 files changed, 4 insertions, 4 deletions

diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py
index 3372f8aea..d20304a1b 100644
--- a/numpy/lib/function_base.py
+++ b/numpy/lib/function_base.py
@@ -524,7 +524,7 @@ def average(a, axis=None, weights=None, returned=False):
     else:
         return avg
 
-def asarray_chkfinite(a):
+def asarray_chkfinite(a, dtype=None, order=None):
     """
     Convert the input to an array, checking for NaNs or Infs.
 
@@ -570,8 +570,8 @@ def asarray_chkfinite(a):
     ``asarray_chkfinite`` is identical to ``asarray``.
 
     >>> a = [1, 2]
-    >>> np.asarray_chkfinite(a)
-    array([1, 2])
+    >>> np.asarray_chkfinite(a, dtype=float)
+    array([1., 2.])
 
     Raises ValueError if array_like contains Nans or Infs.
 
@@ -584,7 +584,7 @@ def asarray_chkfinite(a):
     ValueError
 
     """
-    a = asarray(a)
+    a = asarray(a, dtype=dtype, order=order)
     if (a.dtype.char in typecodes['AllFloat']) \
            and (_nx.isnan(a).any() or _nx.isinf(a).any()):
         raise ValueError(