From 053e3e096d440802141a624576077fae370baf8a Mon Sep 17 00:00:00 2001 From: Nathaniel Starkman Date: Tue, 30 Jan 2024 23:16:38 -0500 Subject: [PATCH] feat(stream): make MockStream class public (#114) * feat(stream): make MockStream class public * test: add to api Signed-off-by: nstarman --- src/galax/dynamics/mockstream/__init__.py | 4 +++- tests/smoke/dynamics/mockstream/test_package.py | 6 ++++-- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/src/galax/dynamics/mockstream/__init__.py b/src/galax/dynamics/mockstream/__init__.py index 9dcfc958..f2a15652 100644 --- 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 index cc14e3a3..3ca20857 100644 --- 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__ )