refactor: organize submodules and tests by spec. (#4)

* refactor: organize submodules and tests by spec.

* Rename submodules to match test file names.

* Adjust for new names.

src/ragged/common/__init__.py

@@ -9,7 +9,15 @@
 
 from __future__ import annotations
 
-from ._creation import (
+from ._spec_array_object import array
+from ._spec_constants import (
+    e,
+    inf,
+    nan,
+    newaxis,
+    pi,
+)
+from ._spec_creation_functions import (
     arange,
     asarray,
     empty,
@@ -27,15 +35,15 @@
     zeros,
     zeros_like,
 )
-from ._datatype import (
+from ._spec_data_type_functions import (
     astype,
     can_cast,
     finfo,
     iinfo,
     isdtype,
     result_type,
 )
-from ._elementwise import (  # pylint: disable=W0622
+from ._spec_elementwise_functions import (  # pylint: disable=W0622
     abs,
     acos,
     acosh,
@@ -96,16 +104,16 @@
     tanh,
     trunc,
 )
-from ._indexing import (
+from ._spec_indexing_functions import (
     take,
 )
-from ._linalg import (
+from ._spec_linear_algebra_functions import (
     matmul,
     matrix_transpose,
     tensordot,
     vecdot,
 )
-from ._manipulation import (
+from ._spec_manipulation_functions import (
     broadcast_arrays,
     broadcast_to,
     concat,
@@ -117,24 +125,23 @@
     squeeze,
     stack,
 )
-from ._obj import array
-from ._search import (
+from ._spec_searching_functions import (
     argmax,
     argmin,
     nonzero,
     where,
 )
-from ._set import (
+from ._spec_set_functions import (
     unique_all,
     unique_counts,
     unique_inverse,
     unique_values,
 )
-from ._sorting import (
+from ._spec_sorting_functions import (
     argsort,
     sort,
 )
-from ._statistical import (  # pylint: disable=W0622
+from ._spec_statistical_functions import (  # pylint: disable=W0622
     max,
     mean,
     min,
@@ -143,13 +150,21 @@
     sum,
     var,
 )
-from ._utility import (  # pylint: disable=W0622
+from ._spec_utility_functions import (  # pylint: disable=W0622
     all,
     any,
 )
 
 __all__ = [
-    # _creation
+    # _spec_array_object
+    "array",
+    # _spec_constants
+    "e",
+    "inf",
+    "nan",
+    "newaxis",
+    "pi",
+    # _spec_creation_functions
     "arange",
     "asarray",
     "empty",
@@ -166,14 +181,14 @@
     "triu",
     "zeros",
     "zeros_like",
-    # _datatype
+    # _spec_data_type_functions
     "astype",
     "can_cast",
     "finfo",
     "iinfo",
     "isdtype",
     "result_type",
-    # _elementwise
+    # _spec_elementwise_functions
     "abs",
     "acos",
     "acosh",
@@ -233,14 +248,14 @@
     "tan",
     "tanh",
     "trunc",
-    # _indexing
+    # _spec_indexing_functions
     "take",
-    # _linalg
+    # _spec_linear_algebra_functions
     "matmul",
     "matrix_transpose",
     "tensordot",
     "vecdot",
-    # _manipulation
+    # _spec_manipulation_functions
     "broadcast_arrays",
     "broadcast_to",
     "concat",
@@ -251,30 +266,28 @@
     "roll",
     "squeeze",
     "stack",
-    # _obj
-    "array",
-    # _search
+    # _spec_searching_functions
     "argmax",
     "argmin",
     "nonzero",
     "where",
-    # _set
+    # _spec_set_functions
     "unique_all",
     "unique_counts",
     "unique_inverse",
     "unique_values",
-    # _sorting
+    # _spec_sorting_functions
     "argsort",
     "sort",
-    # _statistical
+    # _spec_statistical_functions
     "max",
     "mean",
     "min",
     "prod",
     "std",
     "sum",
     "var",
-    # _utility
+    # _spec_utility_functions
     "all",
     "any",
 ]

src/ragged/common/_creation.py → src/ragged/common/_spec_creation_functions.py

@@ -8,7 +8,7 @@
 
 import awkward as ak
 
-from ._obj import array
+from ._spec_array_object import array
 from ._typing import (
     Device,
     Dtype,

src/ragged/common/_datatype.py → src/ragged/common/_spec_data_type_functions.py

@@ -10,7 +10,7 @@
 
 import numpy as np
 
-from ._obj import array
+from ._spec_array_object import array
 from ._typing import Dtype
 
 

src/ragged/common/_elementwise.py → src/ragged/common/_spec_elementwise_functions.py

@@ -6,7 +6,7 @@
 
 from __future__ import annotations
 
-from ._obj import array
+from ._spec_array_object import array
 
 
 def abs(x: array, /) -> array:  # pylint: disable=W0622

src/ragged/common/_indexing.py → src/ragged/common/_spec_indexing_functions.py

@@ -6,7 +6,7 @@
 
 from __future__ import annotations
 
-from ._obj import array
+from ._spec_array_object import array
 
 
 def take(x: array, indices: array, /, *, axis: None | int = None) -> array:

src/ragged/common/_linalg.py → src/ragged/common/_spec_linear_algebra_functions.py

@@ -8,7 +8,7 @@
 
 from collections.abc import Sequence
 
-from ._obj import array
+from ._spec_array_object import array
 
 
 def matmul(x1: array, x2: array, /) -> array:

src/ragged/common/_manipulation.py → src/ragged/common/_spec_manipulation_functions.py

@@ -6,7 +6,7 @@
 
 from __future__ import annotations
 
-from ._obj import array
+from ._spec_array_object import array
 
 
 def broadcast_arrays(*arrays: array) -> list[array]:

src/ragged/common/_search.py → src/ragged/common/_spec_searching_functions.py

@@ -6,7 +6,7 @@
 
 from __future__ import annotations
 
-from ._obj import array
+from ._spec_array_object import array
 
 
 def argmax(x: array, /, *, axis: None | int = None, keepdims: bool = False) -> array:

src/ragged/common/_set.py → src/ragged/common/_spec_set_functions.py

@@ -8,7 +8,7 @@
 
 from collections import namedtuple
 
-from ._obj import array
+from ._spec_array_object import array
 
 unique_all_result = namedtuple(  # pylint: disable=C0103
     "unique_all_result", ["values", "indices", "inverse_indices", "counts"]

src/ragged/common/_sorting.py → src/ragged/common/_spec_sorting_functions.py

@@ -6,7 +6,7 @@
 
 from __future__ import annotations
 
-from ._obj import array
+from ._spec_array_object import array
 
 
 def argsort(

src/ragged/common/_statistical.py → src/ragged/common/_spec_statistical_functions.py

@@ -6,7 +6,7 @@
 
 from __future__ import annotations
 
-from ._obj import array
+from ._spec_array_object import array
 from ._typing import Dtype
 
 

src/ragged/common/_utility.py → src/ragged/common/_spec_utility_functions.py

@@ -6,7 +6,7 @@
 
 from __future__ import annotations
 
-from ._obj import array
+from ._spec_array_object import array
 
 
 def all(  # pylint: disable=W0622