DOC: Fix records doctests by improving docstrings

seberg · seberg · commit f3bde7c100e4 · 2023-06-19T15:18:15.000+02:00
diff --git a/numpy/core/records.py b/numpy/core/records.py
@@ -816,7 +816,7 @@ def fromstring(datastring, dtype=None, shape=None, offset=0, formats=None,
 
     >>> s = '\x01\x02\x03abc'
     >>> np.core.records.fromstring(s, dtype='u1,u1,u1,S3')
-    Traceback (most recent call last)
+    Traceback (most recent call last):
        ...
     TypeError: a bytes-like object is required, not 'str'
     """
@@ -881,18 +881,21 @@ def fromfile(fd, dtype=None, shape=None, offset=0, formats=None,
     >>> from tempfile import TemporaryFile
     >>> a = np.empty(10,dtype='f8,i4,a5')
     >>> a[5] = (0.5,10,'abcde')
-    >>>
-    >>> fd=TemporaryFile()
-    >>> a = a.newbyteorder('<')
-    >>> a.tofile(fd)
-    >>>
-    >>> _ = fd.seek(0)
-    >>> r=np.core.records.fromfile(fd, formats='f8,i4,a5', shape=10,
-    ... byteorder='<')
+
+    Save and load from a temporary file enforcing a little endian byte-order.
+
+    >>> with TemporaryFile() as fd:
+    ...     a = a.newbyteorder('<')
+    ...     a.tofile(fd)
+    ...
+    ...     _ = fd.seek(0)
+    ...     r = np.core.records.fromfile(fd, formats='f8,i4,a5', shape=10,
+    ...                                  byteorder='<')
     >>> print(r[5])
     (0.5, 10, 'abcde')
     >>> r.shape
     (10,)
+
     """
 
     if dtype is None and formats is None:
