arraykit 1.3.1__tar.gz → 1.5.0__tar.gz

{arraykit-1.3.1/arraykit.egg-info → arraykit-1.5.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: arraykit
-Version: 1.3.1
+Version: 1.5.0
 Summary: Array utilities for StaticFrame
 Author: Christopher Ariza, Brandt Bucher, Charles Burkland
-License-Expression: MIT
+License: MIT
 Project-URL: Homepage, https://github.com/static-frame/arraykit
 Keywords: numpy,array
 Classifier: Development Status :: 5 - Production/Stable
@@ -64,6 +64,17 @@ ArrayKit requires the following:
 What is New in ArrayKit
 -------------------------
 
+1.5.0
+............
+Added ``transition_slices_from_group``.
+
+
+1.4.0
+............
+
+Added ``write_array_to_file()``.
+
+
 1.3.1
 ............
 

{arraykit-1.3.1 → arraykit-1.5.0}/README.rst

@@ -35,6 +35,17 @@ ArrayKit requires the following:
 What is New in ArrayKit
 -------------------------
 
+1.5.0
+............
+Added ``transition_slices_from_group``.
+
+
+1.4.0
+............
+
+Added ``write_array_to_file()``.
+
+
 1.3.1
 ............
 

arraykit-1.5.0/VERSION

@@ -0,0 +1,2 @@
+1.5.0
+

{arraykit-1.3.1 → arraykit-1.5.0/arraykit.egg-info}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: arraykit
-Version: 1.3.1
+Version: 1.5.0
 Summary: Array utilities for StaticFrame
 Author: Christopher Ariza, Brandt Bucher, Charles Burkland
-License-Expression: MIT
+License: MIT
 Project-URL: Homepage, https://github.com/static-frame/arraykit
 Keywords: numpy,array
 Classifier: Development Status :: 5 - Production/Stable
@@ -64,6 +64,17 @@ ArrayKit requires the following:
 What is New in ArrayKit
 -------------------------
 
+1.5.0
+............
+Added ``transition_slices_from_group``.
+
+
+1.4.0
+............
+
+Added ``write_array_to_file()``.
+
+
 1.3.1
 ............
 

{arraykit-1.3.1 → arraykit-1.5.0}/arraykit.egg-info/SOURCES.txt

@@ -41,6 +41,7 @@ test/test_nonzero_1d_property.py
 test/test_objectable.py
 test/test_pyi.py
 test/test_split_after_count.py
+test/test_transition_slices_from_group.py
 test/test_tri_map.py
 test/test_type_discovery.py
 test/test_util.py

{arraykit-1.3.1 → arraykit-1.5.0}/pyproject.toml

@@ -19,7 +19,7 @@ authors = [
   { name = "Brandt Bucher" },
   { name = "Charles Burkland" },
 ]
-license = "MIT"
+license = { text = "MIT" }
 keywords = ["numpy", "array"]
 dependencies = [
   "numpy>=1.24.3",
@@ -54,3 +54,23 @@ arraykit = ["__init__.pyi", "py.typed"]
 [tool.setuptools.dynamic]
 version = { file = "VERSION" }
 
+[tool.ruff]
+exclude = [
+    ".git",
+    ".mypy_cache",
+    ".pytest_cache",
+    ".ruff_cache",
+    "build",
+    "dist",
+]
+
+line-length = 90
+indent-width = 4
+
+[tool.ruff.format]
+quote-style = "single"
+indent-style = "space"
+skip-magic-trailing-comma = false
+line-ending = "auto"
+docstring-code-format = true
+docstring-code-line-length = "dynamic"

arraykit-1.5.0/setup.py

@@ -0,0 +1,46 @@
+import typing as tp
+from setuptools import setup, Extension
+from setuptools.command.build_ext import build_ext
+import site, os
+from pathlib import Path
+
+AK_VERSION = Path('VERSION').read_text(encoding='utf-8').strip()
+
+
+def get_ext_dir(*components: tp.Iterable[str]) -> tp.Sequence[str]:
+    dirs = []
+    # Check user site-packages
+    user_site = site.getusersitepackages()
+    if user_site:
+        fp = os.path.join(user_site, *components)
+        if os.path.exists(fp):
+            dirs.append(fp)
+    # Check system site-packages
+    for sp in site.getsitepackages():
+        fp = os.path.join(sp, *components)
+        if os.path.exists(fp):
+            dirs.append(fp)
+    return dirs
+
+
+ext_modules = [
+    Extension(
+        name='arraykit._arraykit',
+        sources=[
+            'src/_arraykit.c',
+            'src/array_go.c',
+            'src/array_to_tuple.c',
+            'src/block_index.c',
+            'src/delimited_to_arrays.c',
+            'src/methods.c',
+            'src/tri_map.c',
+            'src/auto_map.c',
+        ],
+        include_dirs=get_ext_dir('numpy', '_core', 'include') + ['src'],
+        library_dirs=get_ext_dir('numpy', '_core', 'lib'),
+        define_macros=[('AK_VERSION', AK_VERSION)],
+        libraries=['npymath'],
+    )
+]
+
+setup(ext_modules=ext_modules)

{arraykit-1.3.1 → arraykit-1.5.0}/src/__init__.py

@@ -24,6 +24,7 @@ from ._arraykit import delimited_to_arrays as delimited_to_arrays
 from ._arraykit import iterable_str_to_array_1d as iterable_str_to_array_1d
 from ._arraykit import split_after_count as split_after_count
 from ._arraykit import get_new_indexers_and_screen as get_new_indexers_and_screen
+from ._arraykit import write_array_to_file as write_array_to_file
 from ._arraykit import count_iteration as count_iteration
 from ._arraykit import first_true_1d as first_true_1d
 from ._arraykit import first_true_2d as first_true_2d
@@ -32,6 +33,7 @@ from ._arraykit import slice_to_unit as slice_to_unit
 from ._arraykit import array_to_tuple_array as array_to_tuple_array
 from ._arraykit import array_to_tuple_iter as array_to_tuple_iter
 from ._arraykit import nonzero_1d as nonzero_1d
+from ._arraykit import transition_slices_from_group as transition_slices_from_group
 from ._arraykit import is_objectable_dt64 as is_objectable_dt64
 from ._arraykit import is_objectable as is_objectable
 from ._arraykit import astype_array as astype_array

{arraykit-1.3.1 → arraykit-1.5.0}/src/__init__.pyi

@@ -8,22 +8,22 @@ _T = tp.TypeVar('_T')
 __version__: str
 
 _TLabel = tp.Union[
-        tp.Hashable,
-        int,
-        bool,
-        np.bool_,
-        np.integer,
-        float,
-        complex,
-        np.inexact,
-        str,
-        bytes,
-        None,
-        np.datetime64,
-        np.timedelta64,
-        datetime.date,
-        datetime.datetime,
-        tp.Tuple['_TLabel', ...],
+    tp.Hashable,
+    int,
+    bool,
+    np.bool_,
+    np.integer,
+    float,
+    complex,
+    np.inexact,
+    str,
+    bytes,
+    None,
+    np.datetime64,
+    np.timedelta64,
+    datetime.date,
+    datetime.datetime,
+    tp.Tuple['_TLabel', ...],
 ]
 
 class ErrorInitTypeBlocks(RuntimeError):
@@ -31,7 +31,6 @@ class ErrorInitTypeBlocks(RuntimeError):
     def with_traceback(self, tb: Exception) -> Exception: ...
     def __setstate__(self) -> None: ...
 
-
 class NonUniqueError(RuntimeError):
     def __init__(self, *args: tp.Any, **kwargs: tp.Any) -> None: ...
     def with_traceback(self, tb: Exception) -> Exception: ...
@@ -62,20 +61,18 @@ class TriMap:
     def dst_no_fill(self) -> bool: ...
     def map_src_no_fill(self, /, array_from: np.ndarray) -> np.ndarray: ...
     def map_dst_no_fill(self, /, array_from: np.ndarray) -> np.ndarray: ...
-    def map_src_fill(self, /,
-            array_from: np.ndarray,
-            fill_value: tp.Any,
-            fill_value_dtype: np.dtype
-            ) -> np.ndarray: ...
-    def map_dst_fill(self, /,
-            array_from: np.ndarray,
-            fill_value: tp.Any,
-            fill_value_dtype: np.dtype
-            ) -> np.ndarray: ...
-    def map_merge(self, /,
-            array_from_src: np.ndarray,
-            array_from_dst: np.ndarray,
-            ) -> np.ndarray: ...
+    def map_src_fill(
+        self, /, array_from: np.ndarray, fill_value: tp.Any, fill_value_dtype: np.dtype
+    ) -> np.ndarray: ...
+    def map_dst_fill(
+        self, /, array_from: np.ndarray, fill_value: tp.Any, fill_value_dtype: np.dtype
+    ) -> np.ndarray: ...
+    def map_merge(
+        self,
+        /,
+        array_from_src: np.ndarray,
+        array_from_dst: np.ndarray,
+    ) -> np.ndarray: ...
 
 class BlockIndex:
     shape: tp.Tuple[int, int]
@@ -84,46 +81,68 @@ class BlockIndex:
     columns: int
 
     def __init__(
-            block_count: int = 0,
-            row_count: int = -1,
-            bir_count: int = 0,
-            bir_capacity: int = 8,
-            bir_bytes: bytes = b'',
-            dtype: np.dtype = None,
-            ) -> None: ...
+        block_count: int = 0,
+        row_count: int = -1,
+        bir_count: int = 0,
+        bir_capacity: int = 8,
+        bir_bytes: bytes = b'',
+        dtype: np.dtype = None,
+    ) -> None: ...
     def register(self, __value: np.ndarray) -> bool: ...
-    def to_list(self,) -> tp.List[tp.Tuple[int, int]]: ...
-    def to_bytes(self,) -> bytes: ...
-    def copy(self,) -> 'BlockIndex': ...
-    def __len__(self,) -> int: ...
-    def __iter__(self,) -> tp.Iterator[tp.Tuple[int, int]]: ...
-    def __reversed__(self,) -> tp.Iterator[tp.Tuple[int, int]]: ...
+    def to_list(
+        self,
+    ) -> tp.List[tp.Tuple[int, int]]: ...
+    def to_bytes(
+        self,
+    ) -> bytes: ...
+    def copy(
+        self,
+    ) -> 'BlockIndex': ...
+    def __len__(
+        self,
+    ) -> int: ...
+    def __iter__(
+        self,
+    ) -> tp.Iterator[tp.Tuple[int, int]]: ...
+    def __reversed__(
+        self,
+    ) -> tp.Iterator[tp.Tuple[int, int]]: ...
     def __getitem__(self, __key: int) -> tp.Tuple[int, int]: ...
-    def __getstate__(self,) -> tp.Tuple[int, int, int, int, bytes]: ...
+    def __getstate__(
+        self,
+    ) -> tp.Tuple[int, int, int, int, bytes]: ...
     def __setstate__(self, state: tp.Tuple[int, int, int, int, bytes]) -> None: ...
     def get_block(self, __key: int) -> int: ...
     def get_column(self, __key: int) -> int: ...
-    def iter_select(self,
-            __key: tp.Union[slice, np.ndarray, tp.List[int]],
-            ) -> tp.Iterator[tp.Tuple[int, int]]: ...
-    def iter_contiguous(self,
-            __key: tp.Union[slice, np.ndarray, tp.List[int]],
-            *,
-            ascending: bool = False,
-            reduce: bool = False,
-            ) -> tp.Iterator[tp.Tuple[int, tp.Union[slice, int]]]: ...
+    def iter_select(
+        self,
+        __key: tp.Union[slice, np.ndarray, tp.List[int]],
+    ) -> tp.Iterator[tp.Tuple[int, int]]: ...
+    def iter_contiguous(
+        self,
+        __key: tp.Union[slice, np.ndarray, tp.List[int]],
+        *,
+        ascending: bool = False,
+        reduce: bool = False,
+    ) -> tp.Iterator[tp.Tuple[int, tp.Union[slice, int]]]: ...
     def iter_block(self) -> tp.Iterator[tp.Tuple[int, slice]]: ...
 
-
 class FrozenAutoMap:
-    def __init__(self, labels: tp.Iterable[_TLabel] | np.ndarray = (), /,) -> None: ...
-    def get(self, __key: _TLabel, /,) -> int: ...
+    def __init__(
+        self,
+        labels: tp.Iterable[_TLabel] | np.ndarray = (),
+        /,
+    ) -> None: ...
+    def get(
+        self,
+        __key: _TLabel,
+        /,
+    ) -> int: ...
     def keys(self) -> tp.Iterator[_TLabel]: ...
     def items(self) -> tp.Iterator[tuple[_TLabel, int]]: ...
     def values(self) -> tp.Iterator[int]: ...
     def get_all(self, __key: list[_TLabel] | np.ndarray) -> np.ndarray: ...
     def get_any(self, __key: list[_TLabel] | np.ndarray) -> list[int]: ...
-
     def __iter__(self) -> tp.Iterator[_TLabel]: ...
     def __getitem__(self, __key: tp.Any) -> int: ...
     def __contains__(self, __key: tp.Any) -> bool: ...
@@ -133,59 +152,54 @@ class FrozenAutoMap:
     def __getstate__(self) -> tp.Any: ...
     def __setstate__(self, __state: tp.Any) -> None: ...
     def __len__(self) -> int: ...
-
     def __or__(self) -> tp.Any: ...
     def __ror__(self) -> tp.Any: ...
 
-
 class AutoMap(FrozenAutoMap):
-    def __init__(self, labels: tp.Iterable[_TLabel] | np.ndarray = (), /,) -> None: ...
+    def __init__(
+        self,
+        labels: tp.Iterable[_TLabel] | np.ndarray = (),
+        /,
+    ) -> None: ...
     def __ior__(self) -> tp.Any: ...
     def add(self, __key: _TLabel) -> None: ...
     def update(self, __keys: tp.Iterable[_TLabel] | np.ndarray) -> None: ...
 
-
-
-
 def iterable_str_to_array_1d(
-        iterable: tp.Iterable[str],
-        *,
-        dtype: tp.Optional[tp.Any] = None,
-        thousandschar: str = ',',
-        decimalchar: str = '.',
-        ) -> np.ndarray: ...
-
+    iterable: tp.Iterable[str],
+    *,
+    dtype: tp.Optional[tp.Any] = None,
+    thousandschar: str = ',',
+    decimalchar: str = '.',
+) -> np.ndarray: ...
 def delimited_to_arrays(
-        file_like: tp.Iterable[str],
-        *,
-        axis: int = 0,
-        dtypes: tp.Optional[tp.Callable[[int], tp.Any]] = None,
-        line_select: tp.Optional[tp.Callable[[int], bool]] = None,
-        delimiter: str = ',',
-        doublequote: bool = True,
-        escapechar: tp.Optional[str] = '',
-        quotechar: tp.Optional[str] = '"',
-        quoting: int = 0,
-        skipinitialspace: bool = False,
-        strict: bool = False,
-        thousandschar: str = ',',
-        decimalchar: str = '.',
-        ) -> tp.List[np.array]: ...
-
+    file_like: tp.Iterable[str],
+    *,
+    axis: int = 0,
+    dtypes: tp.Optional[tp.Callable[[int], tp.Any]] = None,
+    line_select: tp.Optional[tp.Callable[[int], bool]] = None,
+    delimiter: str = ',',
+    doublequote: bool = True,
+    escapechar: tp.Optional[str] = '',
+    quotechar: tp.Optional[str] = '"',
+    quoting: int = 0,
+    skipinitialspace: bool = False,
+    strict: bool = False,
+    thousandschar: str = ',',
+    decimalchar: str = '.',
+) -> tp.List[np.array]: ...
 def split_after_count(
-        string: str,
-        *,
-        delimiter: str = ',',
-        count: int = 0,
-        doublequote: bool = True,
-        escapechar: tp.Optional[str] = '',
-        quotechar: tp.Optional[str] = '"',
-        quoting: int = 0,
-        strict: bool = False,
-        ) -> tp.Tuple[str, str]: ...
-
+    string: str,
+    *,
+    delimiter: str = ',',
+    count: int = 0,
+    doublequote: bool = True,
+    escapechar: tp.Optional[str] = '',
+    quotechar: tp.Optional[str] = '"',
+    quoting: int = 0,
+    strict: bool = False,
+) -> tp.Tuple[str, str]: ...
 def count_iteration(__iterable: tp.Iterable) -> int: ...
-
 def immutable_filter(__array: np.ndarray) -> np.ndarray: ...
 def mloc(__array: np.ndarray) -> int: ...
 def name_filter(__name: _TLabel) -> _TLabel: ...
@@ -193,20 +207,33 @@ def shape_filter(__array: np.ndarray) -> np.ndarray: ...
 def column_2d_filter(__array: np.ndarray) -> np.ndarray: ...
 def column_1d_filter(__array: np.ndarray) -> np.ndarray: ...
 def row_1d_filter(__array: np.ndarray) -> np.ndarray: ...
-def array_deepcopy(__array: np.ndarray, memo: tp.Optional[tp.Dict[int, tp.Any]]) -> np.ndarray: ...
+def array_deepcopy(
+    __array: np.ndarray, memo: tp.Optional[tp.Dict[int, tp.Any]]
+) -> np.ndarray: ...
 def resolve_dtype(__d1: np.dtype, __d2: np.dtype) -> np.dtype: ...
 def resolve_dtype_iter(__dtypes: tp.Iterable[np.dtype]) -> np.dtype: ...
 def isna_element(__value: tp.Any, include_none: bool = True) -> bool: ...
 def dtype_from_element(__value: tp.Optional[_TLabel]) -> np.dtype: ...
-def get_new_indexers_and_screen(indexers: np.ndarray, positions: np.ndarray) -> tp.Tuple[np.ndarray, np.ndarray]: ...
-
+def get_new_indexers_and_screen(
+    indexers: np.ndarray, positions: np.ndarray
+) -> tp.Tuple[np.ndarray, np.ndarray]: ...
+def write_array_to_file(
+    __array: np.ndarray,
+    __file: tp.IO[bytes],
+    *,
+    fortran_order: bool = False,
+    buffersize: int = 8192,
+) -> None: ...
 def first_true_1d(__array: np.ndarray, *, forward: bool) -> int: ...
 def first_true_2d(__array: np.ndarray, *, forward: bool, axis: int) -> np.ndarray: ...
 def nonzero_1d(__array: np.ndarray, /) -> np.ndarray: ...
+def transition_slices_from_group(
+    __group: np.ndarray, /
+) -> tp.Tuple[tp.Iterator[slice], bool]: ...
 def is_objectable_dt64(__array: np.ndarray, /) -> bool: ...
 def is_objectable(__array: np.ndarray, /) -> bool: ...
 def astype_array(__array: np.ndarray, __dtype: np.dtype | None, /) -> np.ndarray: ...
 def slice_to_ascending_slice(__slice: slice, __size: int) -> slice: ...
 def slice_to_unit(__slice: slice, /) -> int: ...
 def array_to_tuple_array(__array: np.ndarray) -> np.ndarray: ...
-def array_to_tuple_iter(__array: np.ndarray) -> tp.Iterator[tp.Tuple[tp.Any, ...]]: ...
+def array_to_tuple_iter(__array: np.ndarray) -> tp.Iterator[tp.Tuple[tp.Any, ...]]: ...

{arraykit-1.3.1 → arraykit-1.5.0}/src/_arraykit.c

@@ -53,6 +53,7 @@ static PyMethodDef arraykit_methods[] =  {
             NULL},
     {"count_iteration", count_iteration, METH_O, NULL},
     {"nonzero_1d", nonzero_1d, METH_O, NULL},
+    {"transition_slices_from_group", transition_slices_from_group, METH_O, NULL},
     {"is_objectable_dt64", is_objectable_dt64, METH_O, NULL},
     {"is_objectable", is_objectable, METH_O, NULL},
     {"astype_array", astype_array, METH_VARARGS, NULL},
@@ -65,6 +66,10 @@ static PyMethodDef arraykit_methods[] =  {
             (PyCFunction)get_new_indexers_and_screen,
             METH_VARARGS | METH_KEYWORDS,
             NULL},
+    {"write_array_to_file",
+            (PyCFunction)write_array_to_file,
+            METH_VARARGS | METH_KEYWORDS,
+            NULL},
     {NULL},
 };