Fix package (#65)

src/foapy/__init__.py

@@ -14,3 +14,53 @@
     __version__ = "unknown"
 finally:
     del version, PackageNotFoundError
+
+
+# We first need to detect if we're being called as part of the numpy setup
+# procedure itself in a reliable manner.
+try:
+    __FOAPY_SETUP__
+except NameError:
+    __FOAPY_SETUP__ = False
+
+if __FOAPY_SETUP__:
+    sys.stderr.write("Running from numpy source directory.\n")
+else:
+    from .alphabet import alphabet  # noqa: F401
+    from .constants_intervals import binding, mode  # noqa: F401
+    from .intervals import intervals  # noqa: F401
+    from .order import order  # noqa: F401
+
+    # public submodules are imported lazily, therefore are accessible from
+    # __getattr__. Note that `distutils` (deprecated) and `array_api`
+    # (experimental label) are not added here, because `from numpy import *`
+    # must not raise any warnings - that's too disruptive.
+    __foapy_submodules__ = {"ma"}
+
+    __all__ = list(
+        __foapy_submodules__
+        | {"order", "intervals", "exceptions", "alphabet", "binding", "mode"}
+        | {"__version__", "__array_namespace_info__"}
+    )
+
+    def __getattr__(attr):
+        if attr == "exceptions":
+            from . import exceptions
+
+            return exceptions
+        elif attr == "ma":
+            from . import ma
+
+            return ma
+
+        raise AttributeError(
+            "module {!r} has no attribute " "{!r}".format(__name__, attr)
+        )
+
+    def __dir__():
+        public_symbols = globals().keys() | __foapy_submodules__
+        public_symbols -= {
+            "ma",
+            "version",
+        }
+        return list(public_symbols)

src/foapy/intervals.py

@@ -1,6 +1,6 @@
 import numpy as np
 
-from foapy.constants_intervals import binding, mode
+from foapy import binding, mode
 
 
 def intervals(X, bind, mod):

src/foapy/ma/intervals.py

@@ -1,7 +1,7 @@
 import numpy as np
 from numpy import ma
 
-from foapy.constants_intervals import binding, mode
+from foapy import binding, mode
 from foapy.exceptions import InconsistentOrderException, Not1DArrayException
 
 

src/foapy/ma/order.py

@@ -1,9 +1,9 @@
 import numpy as np
 import numpy.ma as ma
 
+from foapy import order as general_order
 from foapy.exceptions import Not1DArrayException
 from foapy.ma.alphabet import alphabet
-from foapy.order import order as general_order
 
 
 def order(X, return_alphabet=False) -> np.ma.MaskedArray:

tests/test_alphabet.py

@@ -4,7 +4,7 @@
 import pytest
 from numpy.testing import assert_array_equal
 
-from foapy.alphabet import alphabet
+from foapy import alphabet
 from foapy.exceptions import Not1DArrayException
 
 

tests/test_intervals.py

@@ -4,8 +4,7 @@
 import pytest
 from numpy.testing import assert_array_equal
 
-from foapy.constants_intervals import binding, mode
-from foapy.intervals import intervals
+from foapy import binding, intervals, mode
 
 
 class TestIntervals(TestCase):

tests/test_order.py

@@ -4,8 +4,8 @@
 import pytest
 from numpy.testing import assert_array_equal
 
+from foapy import order
 from foapy.exceptions import Not1DArrayException
-from foapy.order import order
 
 
 class TestOrder(TestCase):