move test to tests

Joshix-1 · Jun 29, 2024 · 45d132d · 45d132d
1 parent 6e06bb9
commit 45d132d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/tests/__main__.py b/tests/__main__.py
@@ -30,6 +30,7 @@
     StreamEmptyError,
     StreamFinishedError,
     StreamIndexError,
+    _impl,
 )
 from typed_stream._impl._iteration_utils import (
     IterWithCleanUp,
@@ -57,6 +58,11 @@
     noop,
 )
 
+for export in Stream(_impl.__all__).map(partial(getattr, _impl)):  # noqa: F405
+    # pylint: disable-next=protected-access
+    assert export._module() == "typed_stream", f"{export!r}"  # nosec: B101
+    del export
+
 
 def testmod(
     mod: types.ModuleType,

diff --git a/typed_stream/_impl/__init__.py b/typed_stream/_impl/__init__.py
@@ -14,8 +14,3 @@
     *stream.__all__,
     *file_streams.__all__,
 )
-
-for export in Stream(__all__).map(globals().__getitem__):  # noqa: F405
-    # pylint: disable-next=protected-access
-    assert export._module() == "typed_stream", f"{export!r}"  # nosec: B101
-    del export