test: smoke api

Signed-off-by: nstarman <nstarman@users.noreply.github.com>
GalacticDynamics · Jan 31, 2024 · 70bbc0a · 70bbc0a
1 parent 52b4cb4
commit 70bbc0a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/smoke/integrate/test_package.py b/tests/smoke/integrate/test_package.py
@@ -1,9 +1,11 @@
 """Test the :mod:`galax.integrate` module."""
 
 from galax import integrate
-from galax.integrate import _base, _builtin
+from galax.integrate import _api, _base, _builtin
 
 
 def test_all() -> None:
     """Test the API."""
-    assert set(integrate.__all__) == set(_base.__all__ + _builtin.__all__)
+    assert set(integrate.__all__) == set(
+        _api.__all__ + _base.__all__ + _builtin.__all__
+    )