numpy-cache 0.1.0__tar.gz → 0.1.2__tar.gz

{numpy_cache-0.1.0/src/numpy_cache.egg-info → numpy_cache-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: numpy-cache
-Version: 0.1.0
+Version: 0.1.2
 Summary: High-performance LZ4 cache for NumPy arrays
 Author-email: "@macht1212" <nasimov.alexander@gmail.com>
 License-Expression: Apache-2.0
@@ -18,6 +18,7 @@ Requires-Python: <4.0.0,>=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy<3.0.0,>=2.5.2
+Requires-Dist: click<9.0.0,>=8.4.2
 Dynamic: license-file
 
 # NumPy Cache – Fast LZ4-Based Caching for NumPy Arrays
@@ -132,7 +133,8 @@ save(slice_arr, 'slice.npc')  # Handles automatically
 ## 📊 Benchmarks
 ### Test system:
 
-- Ubuntu 24.04.4 LTS, 12th Gen Intel i5-1235U (12 cores), 16 GB RAM, NVMe SSD
+- Ubuntu 24.04.4 LTS, 12th Gen Intel i5-1235U (10 cores), 16 GB RAM, SSD
+- MacOS 15.7.2, Apple Silicon M1 (8 cores), 8 GB RAM, SSD  
 - Python 3.12, NumPy 2.5.2, LZ4 1.9.4
 
 All arrays are `float32`. Sizes:
@@ -142,25 +144,25 @@ All arrays are `float32`. Sizes:
 
 ### Write Performance (time in μs)
 
-| Method| 	0.04 MB| 	1 MB| 
-|-------|----------|--------|
-| `np.save`| 	89.6| 	879.6| 
-| `np.savez`| 	117.4| 	1 083.0| 
-| `np.savez_compressed`| 	1 056.1| 	28 921.5| 
-| **numpy_cache (accel=1)**| 	**71.8**| 	**759.4**| 
-| **numpy_cache (accel=4)**| 	**70.1**| 	**755.6**| 
-| **numpy_cache (accel=16)**| 	**74.9**| 	**738.9**| 
+| Method| Intel i5 <br>	0.04 MB| Intel i5 <br>	1 MB|  Apple M1 <br> 0.04 MB|  Apple M1 <br>	1 MB|
+|-------|----------|--------|--------|------|
+| `np.save`| 	89.6| 	879.6| 86.2 | 548.1 |
+| `np.savez`| 	117.4| 	1 083.0| 94.5 | 554.4 |
+| `np.savez_compressed`| 	1 056.1| 	28 921.5| 1 034.0 | 32 211.2 |
+| **numpy_cache (accel=1)**| 	**71.8**| 	**759.4**| **67.8** | **770.1** |
+| **numpy_cache (accel=4)**| 	**70.1**| 	**755.6**| **64.9** | **635.4** |
+| **numpy_cache (accel=16)**| 	**74.9**| 	**738.9**| **68.0** | **655.2** |
 
 ### Read Performance (time in μs)
 
-| Method| 	0.04 MB| 	1 MB| 
-|-------|----------|--------|
-| `np.save`| 	41.7| 	91.9| 
-| `np.savez`| 	83.0| 	401.1| 
-| `np.savez_compressed`| 	267.5| 	4 858.8| 
-| **numpy_cache (accel=1)**| 	**19.2**| 	**385.0**| 
-| **numpy_cache (accel=4)**| 	**17.5**| 	**421.0**| 
-| **numpy_cache (accel=16)**| 	**17.0**| 	**474.5**| 
+| Method| Intel i5 <br>	0.04 MB| Intel i5 <br>	1 MB|  Apple M1 <br> 0.04 MB|  Apple M1 <br>	1 MB|
+|-------|----------|--------|--------|------|
+| `np.save`| 	41.7| 	91.9| 44.8 | 77.3 |
+| `np.savez`| 	83.0| 	401.1| 95.1 | 225.2 |
+| `np.savez_compressed`| 	267.5| 	4 858.8| 193.7 | 2 876.7 |
+| **numpy_cache (accel=1)**| 	**19.2**| 	**385.0**| **32.7** | **327.1** |
+| **numpy_cache (accel=4)**| 	**17.5**| 	**421.0**| **23.8** | **665.8** |
+| **numpy_cache (accel=16)**| 	**17.0**| 	**474.5**| **19.7** | **187.1** |
 
 ### File Size Comparison (1 MB array)
 - `np.save` / `np.savez`: ~1.0 MB

{numpy_cache-0.1.0 → numpy_cache-0.1.2}/README.md

@@ -110,7 +110,8 @@ save(slice_arr, 'slice.npc')  # Handles automatically
 ## 📊 Benchmarks
 ### Test system:
 
-- Ubuntu 24.04.4 LTS, 12th Gen Intel i5-1235U (12 cores), 16 GB RAM, NVMe SSD
+- Ubuntu 24.04.4 LTS, 12th Gen Intel i5-1235U (10 cores), 16 GB RAM, SSD
+- MacOS 15.7.2, Apple Silicon M1 (8 cores), 8 GB RAM, SSD  
 - Python 3.12, NumPy 2.5.2, LZ4 1.9.4
 
 All arrays are `float32`. Sizes:
@@ -120,25 +121,25 @@ All arrays are `float32`. Sizes:
 
 ### Write Performance (time in μs)
 
-| Method| 	0.04 MB| 	1 MB| 
-|-------|----------|--------|
-| `np.save`| 	89.6| 	879.6| 
-| `np.savez`| 	117.4| 	1 083.0| 
-| `np.savez_compressed`| 	1 056.1| 	28 921.5| 
-| **numpy_cache (accel=1)**| 	**71.8**| 	**759.4**| 
-| **numpy_cache (accel=4)**| 	**70.1**| 	**755.6**| 
-| **numpy_cache (accel=16)**| 	**74.9**| 	**738.9**| 
+| Method| Intel i5 <br>	0.04 MB| Intel i5 <br>	1 MB|  Apple M1 <br> 0.04 MB|  Apple M1 <br>	1 MB|
+|-------|----------|--------|--------|------|
+| `np.save`| 	89.6| 	879.6| 86.2 | 548.1 |
+| `np.savez`| 	117.4| 	1 083.0| 94.5 | 554.4 |
+| `np.savez_compressed`| 	1 056.1| 	28 921.5| 1 034.0 | 32 211.2 |
+| **numpy_cache (accel=1)**| 	**71.8**| 	**759.4**| **67.8** | **770.1** |
+| **numpy_cache (accel=4)**| 	**70.1**| 	**755.6**| **64.9** | **635.4** |
+| **numpy_cache (accel=16)**| 	**74.9**| 	**738.9**| **68.0** | **655.2** |
 
 ### Read Performance (time in μs)
 
-| Method| 	0.04 MB| 	1 MB| 
-|-------|----------|--------|
-| `np.save`| 	41.7| 	91.9| 
-| `np.savez`| 	83.0| 	401.1| 
-| `np.savez_compressed`| 	267.5| 	4 858.8| 
-| **numpy_cache (accel=1)**| 	**19.2**| 	**385.0**| 
-| **numpy_cache (accel=4)**| 	**17.5**| 	**421.0**| 
-| **numpy_cache (accel=16)**| 	**17.0**| 	**474.5**| 
+| Method| Intel i5 <br>	0.04 MB| Intel i5 <br>	1 MB|  Apple M1 <br> 0.04 MB|  Apple M1 <br>	1 MB|
+|-------|----------|--------|--------|------|
+| `np.save`| 	41.7| 	91.9| 44.8 | 77.3 |
+| `np.savez`| 	83.0| 	401.1| 95.1 | 225.2 |
+| `np.savez_compressed`| 	267.5| 	4 858.8| 193.7 | 2 876.7 |
+| **numpy_cache (accel=1)**| 	**19.2**| 	**385.0**| **32.7** | **327.1** |
+| **numpy_cache (accel=4)**| 	**17.5**| 	**421.0**| **23.8** | **665.8** |
+| **numpy_cache (accel=16)**| 	**17.0**| 	**474.5**| **19.7** | **187.1** |
 
 ### File Size Comparison (1 MB array)
 - `np.save` / `np.savez`: ~1.0 MB

{numpy_cache-0.1.0 → numpy_cache-0.1.2}/csrc/cache_module.c

@@ -140,6 +140,84 @@ static int read_cache_file(const char* path, CacheHeader* header, void** compres
   return 0;
 }
 
+static PyObject* shape_to_tuple(const CacheHeader* header) {
+    PyObject* shape_tuple = PyTuple_New(header->ndim);
+    if (!shape_tuple) return NULL;
+    
+    for (uint32_t i = 0; i < header->ndim; ++i) {
+        PyObject* item = PyLong_FromUnsignedLongLong(header->shape[i]);
+        if (!item) {
+            Py_DECREF(shape_tuple);
+            return NULL;
+        }
+        PyTuple_SetItem(shape_tuple, i, item);
+    }
+    return shape_tuple;
+}
+
+static PyArrayObject* create_array_from_header(const CacheHeader* header, void* data) {
+    PyArray_Descr* descr = PyArray_DescrFromType((int)header->dtype);
+    if (!descr) {
+        PyErr_SetString(PyExc_RuntimeError, "Unsupported dtype");
+        return NULL;
+    }
+    
+    npy_intp dims[MAX_DIMS];
+    for (int i = 0; i < (int)header->ndim; ++i) {
+        dims[i] = (npy_intp)header->shape[i];
+    }
+    
+    PyArrayObject* arr = (PyArrayObject*)PyArray_NewFromDescr(
+        &PyArray_Type,
+        descr,
+        (int)header->ndim,
+        dims,
+        NULL,
+        data,
+        NPY_ARRAY_OWNDATA,
+        NULL
+    );
+    
+    if (!arr) {
+        PyErr_SetString(PyExc_RuntimeError, "Failed to create NumPy array");
+    }
+    return arr;
+}
+
+static int validate_header_size(const CacheHeader* header) {
+    if (header->uncompressed_size > MAX_SIZE) {
+        PyErr_SetString(PyExc_RuntimeError, "Uncompressed size too large (>1GB)");
+        return -1;
+    }
+    if (header->uncompressed_size > INT_MAX) {
+        PyErr_SetString(PyExc_RuntimeError, "Uncompressed size too large for LZ4");
+        return -1;
+    }
+    return 0;
+}
+
+static PyObject* header_to_dict(const CacheHeader* header) {
+    PyObject* dict = PyDict_New();
+    if (!dict) return NULL;
+    
+    PyDict_SetItemString(dict, "magic", PyLong_FromUnsignedLong(header->magic));
+    PyDict_SetItemString(dict, "version", PyLong_FromUnsignedLong(header->version));
+    PyDict_SetItemString(dict, "ndim", PyLong_FromUnsignedLong(header->ndim));
+    PyDict_SetItemString(dict, "dtype", PyLong_FromUnsignedLong(header->dtype));
+    PyDict_SetItemString(dict, "uncompressed_size", PyLong_FromUnsignedLongLong(header->uncompressed_size));
+    PyDict_SetItemString(dict, "compressed_size", PyLong_FromUnsignedLongLong(header->compressed_size));
+    
+    PyObject* shape_tuple = shape_to_tuple(header);
+    if (!shape_tuple) {
+        Py_DECREF(dict);
+        return NULL;
+    }
+    PyDict_SetItemString(dict, "shape", shape_tuple);
+    Py_DECREF(shape_tuple);
+    
+    return (PyObject*)dict;
+}
+
 static PyObject* cache_save(PyObject* self, PyObject* args, PyObject* kwargs) {
   PyObject* obj = NULL;
   const char* path = NULL;
@@ -236,93 +314,91 @@ static PyObject* cache_save(PyObject* self, PyObject* args, PyObject* kwargs) {
 }
 
 static PyObject* cache_load(PyObject* self, PyObject* args) {
-  const char* path = NULL;
-  if (!PyArg_ParseTuple(args, "s", &path)) return NULL;
-
-  CacheHeader header;
-  void* compressed_data = NULL;
-
-  if (read_cache_file(path, &header, &compressed_data) != 0) return NULL;
-
-  if (header.uncompressed_size > MAX_SIZE) {
-    PyErr_SetString(PyExc_RuntimeError, "Uncompressed size too large (>1GB)");
-    free(compressed_data);
-    return NULL;
-  }
-
-  if (header.uncompressed_size > INT_MAX) {
-    PyErr_SetString(PyExc_RuntimeError, "Uncompressed size too large for LZ4");
-    free(compressed_data);
-    return NULL;
-  }
-
-  if (header.uncompressed_size == 0) {
-    free(compressed_data);
-    PyArray_Descr* descr = PyArray_DescrFromType((int)header.dtype);
-    if (!descr) {
-        PyErr_SetString(PyExc_RuntimeError, "Unsupported dtype");
+    const char* path = NULL;
+    if (!PyArg_ParseTuple(args, "s", &path)) return NULL;
+    
+    CacheHeader header;
+    void* compressed_data = NULL;
+    
+    if (read_cache_file(path, &header, &compressed_data) != 0) return NULL;
+    
+    if (validate_header_size(&header) != 0) {
+        free(compressed_data);
         return NULL;
     }
-    npy_intp dims[MAX_DIMS];
-    for (int i = 0; i < (int)header.ndim; ++i)
-        dims[i] = (npy_intp)header.shape[i];
-    return (PyObject*)PyArray_Zeros((int)header.ndim, dims, descr, 0);
-  }
-
-  void* decompressed_data = malloc(header.uncompressed_size);
-  if (!decompressed_data) {
-    PyErr_NoMemory();
+    
+    if (header.uncompressed_size == 0) {
+        free(compressed_data);
+        PyArray_Descr* descr = PyArray_DescrFromType((int)header.dtype);
+        if (!descr) {
+            PyErr_SetString(PyExc_RuntimeError, "Unsupported dtype");
+            return NULL;
+        }
+        npy_intp dims[MAX_DIMS];
+        for (int i = 0; i < (int)header.ndim; ++i)
+            dims[i] = (npy_intp)header.shape[i];
+        return (PyObject*)PyArray_Zeros((int)header.ndim, dims, descr, 0);
+    }
+    
+    void* decompressed_data = malloc(header.uncompressed_size);
+    if (!decompressed_data) {
+        PyErr_NoMemory();
+        free(compressed_data);
+        return NULL;
+    }
+    
+    int decompressed_size = LZ4_decompress_safe(
+        (const char*)compressed_data,
+        (char*)decompressed_data,
+        (int)header.compressed_size,
+        (int)header.uncompressed_size
+    );
     free(compressed_data);
-    return NULL;
-  }
-
-  int decompressed_size = LZ4_decompress_safe((const char*)compressed_data,
-                                              (char*)decompressed_data,
-                                              (int)header.compressed_size,
-                                              (int)header.uncompressed_size);
-  free(compressed_data);
-
-  if (decompressed_size < 0) {
-    PyErr_SetString(PyExc_RuntimeError, "LZ4 decompression failed");
-    free(decompressed_data);
-    return NULL;
-  }
-
-  if ((uint64_t)decompressed_size != header.uncompressed_size) {
-    PyErr_SetString(PyExc_RuntimeError, "Decompressed size mismatch");
-    free(decompressed_data);
-    return NULL;
-  }
-
-  PyArray_Descr* descr = PyArray_DescrFromType((int)header.dtype);
-  if (!descr) {
-    PyErr_SetString(PyExc_RuntimeError, "Unsupported dtype");
-    free(decompressed_data);
-    return NULL;
-  }
-
-  npy_intp dims[MAX_DIMS];
-  for (int i = 0; i < (int)header.ndim; ++i)
-    dims[i] = (npy_intp)header.shape[i];
-
-  PyArrayObject* arr = (PyArrayObject*)PyArray_NewFromDescr(
-          &PyArray_Type,
-          descr,
-          (int)header.ndim,
-          dims,
-          NULL,
-          decompressed_data,
-          NPY_ARRAY_OWNDATA,
-          NULL
-  );
-
-  if (!arr) {
-    PyErr_SetString(PyExc_RuntimeError, "Failed to create NumPy array");
-    free(decompressed_data);
-    return NULL;
-  }
+    
+    if (decompressed_size < 0) {
+        PyErr_SetString(PyExc_RuntimeError, "LZ4 decompression failed");
+        free(decompressed_data);
+        return NULL;
+    }
+    
+    if ((uint64_t)decompressed_size != header.uncompressed_size) {
+        PyErr_SetString(PyExc_RuntimeError, "Decompressed size mismatch");
+        free(decompressed_data);
+        return NULL;
+    }
+    
+    PyArrayObject* arr = create_array_from_header(&header, decompressed_data);
+    if (!arr) {
+        free(decompressed_data);
+        return NULL;
+    }
+    
+    return (PyObject*)arr;
+}
 
-  return (PyObject*)arr;
+static PyObject* inspect_header(PyObject* self, PyObject* args) {
+    const char* path = NULL;
+    if (!PyArg_ParseTuple(args, "s", &path)) return NULL;
+    
+    FILE* f = fopen(path, "rb");
+    if (!f) {
+        PyErr_Format(PyExc_IOError, "Cannot open file for reading: %s", strerror(errno));
+        return NULL;
+    }
+    
+    CacheHeader header;
+    if (fread(&header, sizeof(CacheHeader), 1, f) != 1) {
+        PyErr_SetString(PyExc_IOError, "Failed to read header (file too short or corrupt)");
+        fclose(f);
+        return NULL;
+    }
+    
+    if (fclose(f) != 0) {
+        PyErr_Format(PyExc_OSError, "Failed to close file: %s", strerror(errno));
+        return NULL;
+    }
+    
+    return header_to_dict(&header);
 }
 
 static PyMethodDef CacheMethods[] = {
@@ -348,6 +424,10 @@ static PyMethodDef CacheMethods[] = {
     "Raises:\n"
     "  RuntimeError on invalid or corrupted file."
   },
+  {"inspect", inspect_header, METH_VARARGS,
+     "Read and display the header of a cache file without loading the data.\n"
+     "Returns:\n A dictionary with magic, version, ndim, dtype, shape, sizes.\n"
+  },
   {NULL, NULL, 0, NULL}
 };
 

{numpy_cache-0.1.0 → numpy_cache-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "numpy-cache"
-version = "0.1.0"
+version = "0.1.2"
 description = "High-performance LZ4 cache for NumPy arrays"
 authors = [
     {name = "@macht1212", email = "nasimov.alexander@gmail.com"}
@@ -20,6 +20,7 @@ classifiers = [
 ]
 dependencies = [
     "numpy>=2.5.2,<3.0.0",
+    "click (>=8.4.2,<9.0.0)",
 ]
 
 [project.urls]
@@ -40,4 +41,7 @@ build-backend = "setuptools.build_meta"
 
 [tool.setuptools.packages.find]
 where = ["src"]
-include = ["numpy_cache*"]
+include = ["numpy_cache*"]
+
+[tool.poetry.scripts]
+numpy-cache = "numpy_cache.__main__:cli"

{numpy_cache-0.1.0 → numpy_cache-0.1.2}/setup.py

@@ -81,9 +81,14 @@ print("===========================")
 
 setup(
     name="numpy-cache",
-    version="0.1.0",
+    version="0.1.2",
     packages=["numpy_cache"],
     package_dir={"": "src"},
     ext_modules=[cache_module],
     zip_safe=False,
+    entry_points={
+    'console_scripts': [
+        'numpy-cache=numpy_cache.__main__:cli',
+        ],
+    },
 )

numpy_cache-0.1.2/src/numpy_cache/__init__.py

@@ -0,0 +1,29 @@
+from .__main__ import cli
+from ._cache import (
+    ACCELERATION_DEFAULT,
+    ACCELERATION_MAX,
+    ACCELERATION_MIN,
+    MAGIC,
+    MAX_DIMS,
+    MAX_SIZE,
+    VERSION,
+    inspect,
+    load,
+    save,
+)
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "ACCELERATION_DEFAULT",
+    "ACCELERATION_MAX",
+    "ACCELERATION_MIN",
+    "MAGIC",
+    "MAX_DIMS",
+    "MAX_SIZE",
+    "VERSION",
+    "cli",
+    "inspect",
+    "load",
+    "save",
+]

numpy_cache-0.1.2/src/numpy_cache/__main__.py

@@ -0,0 +1,41 @@
+import os
+from dataclasses import dataclass
+
+import click
+
+from ._cache import inspect
+
+
+@dataclass
+class Header:
+    magic: str
+    version: int
+    ndim: int
+    dtype: int
+    uncompressed_size: int
+    compressed_size: int
+    shape: tuple[int, ...]
+
+
+@click.group()
+def cli(): ...
+
+@cli.command(name="inspect")
+@click.argument("path")
+def show_file_headers(path: str):
+    if not os.path.exists(path):
+        raise FileNotFoundError(f"File not find on {path=}.")
+
+    info = Header(**inspect(path))
+
+    click.echo(
+        f"Magic: {info.magic}\nVersion: {info.version}\n"
+        f"NDIM: {info.ndim}\ndType: {info.dtype}\n"
+        f"Uncompressed size: {info.uncompressed_size}\n"
+        f"Compressed size: {info.compressed_size}\n"
+        f"Shape: {info.shape}"
+    )
+
+
+if __name__ == "__main__":
+    cli()

{numpy_cache-0.1.0 → numpy_cache-0.1.2/src/numpy_cache.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: numpy-cache
-Version: 0.1.0
+Version: 0.1.2
 Summary: High-performance LZ4 cache for NumPy arrays
 Author-email: "@macht1212" <nasimov.alexander@gmail.com>
 License-Expression: Apache-2.0
@@ -18,6 +18,7 @@ Requires-Python: <4.0.0,>=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy<3.0.0,>=2.5.2
+Requires-Dist: click<9.0.0,>=8.4.2
 Dynamic: license-file
 
 # NumPy Cache – Fast LZ4-Based Caching for NumPy Arrays
@@ -132,7 +133,8 @@ save(slice_arr, 'slice.npc')  # Handles automatically
 ## 📊 Benchmarks
 ### Test system:
 
-- Ubuntu 24.04.4 LTS, 12th Gen Intel i5-1235U (12 cores), 16 GB RAM, NVMe SSD
+- Ubuntu 24.04.4 LTS, 12th Gen Intel i5-1235U (10 cores), 16 GB RAM, SSD
+- MacOS 15.7.2, Apple Silicon M1 (8 cores), 8 GB RAM, SSD  
 - Python 3.12, NumPy 2.5.2, LZ4 1.9.4
 
 All arrays are `float32`. Sizes:
@@ -142,25 +144,25 @@ All arrays are `float32`. Sizes:
 
 ### Write Performance (time in μs)
 
-| Method| 	0.04 MB| 	1 MB| 
-|-------|----------|--------|
-| `np.save`| 	89.6| 	879.6| 
-| `np.savez`| 	117.4| 	1 083.0| 
-| `np.savez_compressed`| 	1 056.1| 	28 921.5| 
-| **numpy_cache (accel=1)**| 	**71.8**| 	**759.4**| 
-| **numpy_cache (accel=4)**| 	**70.1**| 	**755.6**| 
-| **numpy_cache (accel=16)**| 	**74.9**| 	**738.9**| 
+| Method| Intel i5 <br>	0.04 MB| Intel i5 <br>	1 MB|  Apple M1 <br> 0.04 MB|  Apple M1 <br>	1 MB|
+|-------|----------|--------|--------|------|
+| `np.save`| 	89.6| 	879.6| 86.2 | 548.1 |
+| `np.savez`| 	117.4| 	1 083.0| 94.5 | 554.4 |
+| `np.savez_compressed`| 	1 056.1| 	28 921.5| 1 034.0 | 32 211.2 |
+| **numpy_cache (accel=1)**| 	**71.8**| 	**759.4**| **67.8** | **770.1** |
+| **numpy_cache (accel=4)**| 	**70.1**| 	**755.6**| **64.9** | **635.4** |
+| **numpy_cache (accel=16)**| 	**74.9**| 	**738.9**| **68.0** | **655.2** |
 
 ### Read Performance (time in μs)
 
-| Method| 	0.04 MB| 	1 MB| 
-|-------|----------|--------|
-| `np.save`| 	41.7| 	91.9| 
-| `np.savez`| 	83.0| 	401.1| 
-| `np.savez_compressed`| 	267.5| 	4 858.8| 
-| **numpy_cache (accel=1)**| 	**19.2**| 	**385.0**| 
-| **numpy_cache (accel=4)**| 	**17.5**| 	**421.0**| 
-| **numpy_cache (accel=16)**| 	**17.0**| 	**474.5**| 
+| Method| Intel i5 <br>	0.04 MB| Intel i5 <br>	1 MB|  Apple M1 <br> 0.04 MB|  Apple M1 <br>	1 MB|
+|-------|----------|--------|--------|------|
+| `np.save`| 	41.7| 	91.9| 44.8 | 77.3 |
+| `np.savez`| 	83.0| 	401.1| 95.1 | 225.2 |
+| `np.savez_compressed`| 	267.5| 	4 858.8| 193.7 | 2 876.7 |
+| **numpy_cache (accel=1)**| 	**19.2**| 	**385.0**| **32.7** | **327.1** |
+| **numpy_cache (accel=4)**| 	**17.5**| 	**421.0**| **23.8** | **665.8** |
+| **numpy_cache (accel=16)**| 	**17.0**| 	**474.5**| **19.7** | **187.1** |
 
 ### File Size Comparison (1 MB array)
 - `np.save` / `np.savez`: ~1.0 MB

{numpy_cache-0.1.0 → numpy_cache-0.1.2}/src/numpy_cache.egg-info/SOURCES.txt

@@ -4,6 +4,7 @@ pyproject.toml
 setup.py
 csrc/cache_module.c
 src/numpy_cache/__init__.py
+src/numpy_cache/__main__.py
 src/numpy_cache.egg-info/PKG-INFO
 src/numpy_cache.egg-info/SOURCES.txt
 src/numpy_cache.egg-info/dependency_links.txt

{numpy_cache-0.1.0 → numpy_cache-0.1.2}/src/numpy_cache.egg-info/requires.txt

@@ -1 +1,2 @@
 numpy<3.0.0,>=2.5.2
+click<9.0.0,>=8.4.2

{numpy_cache-0.1.0 → numpy_cache-0.1.2}/tests/test_cache.py

@@ -11,6 +11,7 @@ from numpy_cache._cache import (
     MAX_DIMS,
     MAX_SIZE,
     VERSION,
+    inspect,
     load,
     save,
 )
@@ -217,3 +218,195 @@ def test_metadata_preserved(temp_file):
     assert loaded.shape == (2, 3, 4)
     assert loaded.dtype == np.int16
     np.testing.assert_array_equal(arr, loaded)
+
+def test_inspect_valid_file():
+    arr = np.random.randn(100, 100).astype(np.float32)
+    with tempfile.NamedTemporaryFile(suffix=".npc", delete=False) as f:
+        path = f.name
+
+    try:
+        save(arr, path)
+        info = inspect(path)
+
+        assert isinstance(info, dict)
+        assert "magic" in info
+        assert "version" in info
+        assert "ndim" in info
+        assert "dtype" in info
+        assert "uncompressed_size" in info
+        assert "compressed_size" in info
+        assert "shape" in info
+
+        assert info["magic"] == 0x4C5A4E43  # "LZNC"
+        assert info["version"] == 1
+        assert info["ndim"] == 2
+        assert info["dtype"] == np.dtype(np.float32).num  # 11
+        assert info["uncompressed_size"] >= 0
+        assert info["shape"] == (100, 100)
+        assert info["compressed_size"] > 0
+
+    finally:
+        if os.path.exists(path):
+            os.remove(path)
+
+
+def test_inspect_empty_array():
+    arr = np.array([]).astype(np.float32)
+    with tempfile.NamedTemporaryFile(suffix=".npc", delete=False) as f:
+        path = f.name
+
+    try:
+        save(arr, path)
+        info = inspect(path)
+
+        assert info["ndim"] == 1
+        assert info["shape"] == (0,)
+        assert info["uncompressed_size"] == 0
+        assert info["compressed_size"] == 0
+
+    finally:
+        if os.path.exists(path):
+            os.remove(path)
+
+
+def test_inspect_different_dtypes():
+    dtypes = [
+        np.float32, np.float64,
+        np.int8, np.int16, np.int32, np.int64,
+        np.uint8, np.uint16, np.uint32, np.uint64,
+        np.bool_
+    ]
+
+    for dtype in dtypes:
+        if dtype == np.bool_:
+            arr = np.random.choice([True, False], size=(10, 10))
+        else:
+            arr = np.random.randint(0, 100, size=(10, 10)).astype(dtype)
+
+        with tempfile.NamedTemporaryFile(suffix=".npc", delete=False) as f:
+            path = f.name
+
+        try:
+            save(arr, path)
+            info = inspect(path)
+            assert info["dtype"] == np.dtype(dtype).num
+            assert info["shape"] == (10, 10)
+            assert info["uncompressed_size"] > 0
+        finally:
+            if os.path.exists(path):
+                os.remove(path)
+
+
+def test_inspect_multidimensional():
+    shapes = [(10, 10, 10), (5, 6, 7, 8), (3, 4, 5, 6, 7)]
+
+    for shape in shapes:
+        arr = np.random.randn(*shape).astype(np.float32)
+        with tempfile.NamedTemporaryFile(suffix=".npc", delete=False) as f:
+            path = f.name
+
+        try:
+            save(arr, path)
+            info = inspect(path)
+            assert info["ndim"] == len(shape)
+            assert info["shape"] == shape
+            assert info["uncompressed_size"] == np.prod(shape) * 4
+        finally:
+            if os.path.exists(path):
+                os.remove(path)
+
+
+def test_inspect_non_contiguous():
+    arr = np.random.randn(50, 50).astype(np.float32)
+    sliced = arr[::2, ::2]
+
+    with tempfile.NamedTemporaryFile(suffix=".npc", delete=False) as f:
+        path = f.name
+
+    try:
+        save(sliced, path)
+        info = inspect(path)
+        assert info["shape"] == (25, 25)
+        assert info["uncompressed_size"] == 25 * 25 * 4
+    finally:
+        if os.path.exists(path):
+            os.remove(path)
+
+
+def test_inspect_nonexistent_file():
+    with pytest.raises(IOError, match="Cannot open file for reading"):
+        inspect("/nonexistent/file/path.npc")
+
+
+def test_inspect_invalid_file():
+    with tempfile.NamedTemporaryFile(suffix=".txt", delete=False) as f:
+        path = f.name
+        f.write(b"This is not a numpy-cache file")
+
+    try:
+        with pytest.raises(OSError, match="Failed to read header"):
+            inspect(path)
+    finally:
+        if os.path.exists(path):
+            os.remove(path)
+
+
+def test_inspect_corrupted_header():
+    arr = np.random.randn(10, 10).astype(np.float32)
+    with tempfile.NamedTemporaryFile(suffix=".npc", delete=False) as f:
+        path = f.name
+
+    try:
+        save(arr, path)
+
+        with open(path, "r+b") as f:
+            f.seek(0)
+            f.write(b"\x00\x01\x02\x03")
+
+        info = inspect(path)
+        assert isinstance(info, dict)
+    finally:
+        if os.path.exists(path):
+            os.remove(path)
+
+
+def test_inspect_very_large_ndim():
+    arr = np.random.randn(10, 10).astype(np.float32)
+    with tempfile.NamedTemporaryFile(suffix=".npc", delete=False) as f:
+        path = f.name
+
+    try:
+        save(arr, path)
+
+        with open(path, "r+b") as f:
+            f.seek(88)
+            f.write(b"\xFF\xFF\xFF\xFF")
+
+        try:
+            info = inspect(path)
+            assert info["ndim"] > 1000
+        except (MemoryError, OverflowError, ValueError, SystemError):
+            pass
+
+    finally:
+        if os.path.exists(path):
+            os.remove(path)
+
+
+def test_inspect_acceleration_parameter():
+    accelerations = [1, 4, 16]
+
+    for accel in accelerations:
+        arr = np.random.randn(100, 100).astype(np.float32)
+        with tempfile.NamedTemporaryFile(suffix=".npc", delete=False) as f:
+            path = f.name
+
+        try:
+            save(arr, path, acceleration=accel)
+            info = inspect(path)
+            assert info["shape"] == (100, 100)
+            assert info["dtype"] == np.dtype(np.float32).num
+            assert info["compressed_size"] > 0
+        finally:
+            if os.path.exists(path):
+                os.remove(path)

numpy_cache-0.1.0/src/numpy_cache/__init__.py

@@ -1,25 +0,0 @@
-from ._cache import (
-    ACCELERATION_DEFAULT,
-    ACCELERATION_MAX,
-    ACCELERATION_MIN,
-    MAGIC,
-    MAX_DIMS,
-    MAX_SIZE,
-    VERSION,
-    load,
-    save,
-)
-
-__version__ = "0.1.0"
-
-__all__ = [
-    "ACCELERATION_DEFAULT", 
-    "ACCELERATION_MAX", 
-    "ACCELERATION_MIN", 
-    "MAGIC", 
-    "MAX_DIMS", 
-    "MAX_SIZE", 
-    "VERSION", 
-    "load", 
-    "save"
-]