feat(stream): make MockStream class public (#114)

* feat(stream): make MockStream class public * test: add to api Signed-off-by: nstarman <nstarman@users.noreply.github.com>
GalacticDynamics · Jan 31, 2024 · 053e3e0 · 053e3e0
1 parent f692ac8
commit 053e3e0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/galax/dynamics/mockstream/__init__.py b/src/galax/dynamics/mockstream/__init__.py
@@ -1,9 +1,11 @@
 """galax: Galactic Dynamix in Jax."""
 
-from . import _df, _mockstream_generator
+from . import _core, _df, _mockstream_generator
+from ._core import *
 from ._df import *
 from ._mockstream_generator import *
 
 __all__: list[str] = []
 __all__ += _df.__all__
+__all__ += _core.__all__
 __all__ += _mockstream_generator.__all__
diff --git a/tests/smoke/dynamics/mockstream/test_package.py b/tests/smoke/dynamics/mockstream/test_package.py
@@ -3,8 +3,10 @@
 from galax.dynamics import mockstream
 
 
-def test_all():
+def test_all() -> None:
     """Test the `galax.dynamics.mockstream` API."""
     assert set(mockstream.__all__) == set(
-        mockstream._df.__all__ + mockstream._mockstream_generator.__all__
+        mockstream._df.__all__
+        + mockstream._core.__all__
+        + mockstream._mockstream_generator.__all__
     )