Adjust mypy tests

mtsokol · mtsokol · commit 88885c5f8d79 · 2023-08-31T21:42:28.000+02:00
diff --git a/doc/source/reference/routines.io.rst b/doc/source/reference/routines.io.rst
@@ -89,10 +89,3 @@ Binary format description
    :toctree: generated/
 
    lib.format
-
-Object wrappers
----------------
-.. autosummary::
-   :toctree: generated/
-
-   lib.npyio.BagObj
diff --git a/numpy/lib/_npyio_impl.py b/numpy/lib/_npyio_impl.py
@@ -35,7 +35,6 @@
     overrides.array_function_dispatch, module='numpy')
 
 
-@set_module('numpy.lib.npyio')
 class BagObj:
     """
     BagObj(obj)
@@ -49,7 +48,7 @@ class BagObj:
 
     Examples
     --------
-    >>> from numpy.lib.npyio import BagObj as BO
+    >>> from numpy.lib._npyio_impl import BagObj as BO
     >>> class BagDemo:
     ...     def __getitem__(self, key): # An instance of BagObj(BagDemo)
     ...                                 # will call this method when any
diff --git a/numpy/lib/npyio.py b/numpy/lib/npyio.py
@@ -1 +1 @@
-from ._npyio_impl import BagObj, DataSource, NpzFile
+from ._npyio_impl import DataSource, NpzFile
diff --git a/numpy/lib/npyio.pyi b/numpy/lib/npyio.pyi
@@ -1,5 +1,4 @@
 from numpy.lib._npyio_impl import (
-    BagObj as BagObj,
     DataSource as DataSource,
     NpzFile as NpzFile,
 )
diff --git a/numpy/typing/tests/data/reveal/npyio.pyi b/numpy/typing/tests/data/reveal/npyio.pyi
@@ -10,7 +10,6 @@ pathlib_path: pathlib.Path
 str_file: IO[str]
 bytes_file: IO[bytes]
 
-bag_obj: np.lib.npyio.BagObj[int]
 npz_file: np.lib.npyio.NpzFile
 
 AR_i8: npt.NDArray[np.int64]
@@ -26,15 +25,12 @@ class BytesReader:
 bytes_writer: BytesWriter
 bytes_reader: BytesReader
 
-reveal_type(bag_obj.a)  # E: int
-reveal_type(bag_obj.b)  # E: int
-
 reveal_type(npz_file.zip)  # E: zipfile.ZipFile
 reveal_type(npz_file.fid)  # E: Union[None, typing.IO[builtins.str]]
 reveal_type(npz_file.files)  # E: list[builtins.str]
 reveal_type(npz_file.allow_pickle)  # E: bool
 reveal_type(npz_file.pickle_kwargs)  # E: Union[None, typing.Mapping[builtins.str, Any]]
-reveal_type(npz_file.f)  # E: lib.npyio.BagObj[lib.npyio.NpzFile]
+reveal_type(npz_file.f)  # E: lib._npyio_impl.BagObj[lib.npyio.NpzFile]
 reveal_type(npz_file["test"])  # E: ndarray[Any, dtype[Any]]
 reveal_type(len(npz_file))  # E: int
 with npz_file as f:

Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-from ._npyio_impl import BagObj, DataSource, NpzFile`
	`1`	`+from ._npyio_impl import DataSource, NpzFile`
Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,4 @@`
`1`	`1`	`from numpy.lib._npyio_impl import (`
`2`		`- BagObj as BagObj,`
`3`	`2`	`DataSource as DataSource,`
`4`	`3`	`NpzFile as NpzFile,`
`5`	`4`	`)`