imrw 0.1.0__tar.gz

imrw-0.1.0/PKG-INFO

@@ -0,0 +1,5 @@
+Metadata-Version: 2.4
+Name: imrw
+Version: 0.1.0
+Requires-Dist: numpy
+Requires-Dist: pillow

imrw-0.1.0/README.md

@@ -0,0 +1,20 @@
+# imrw
+Minimal image I/O library for Python.
+
+## Features
+- Read/Write images via `Pillow` & `numpy`.
+- Clean and simple interface.
+- `imread` always returns an RGB (`H x W x 3`) numpy array.
+- `imwrite` expects a `uint8` array with shape `H x W`, `H x W x 1`, `H x W x 3`, or `H x W x 4`.
+
+## Quick Start
+```bash
+pip install imrw
+```
+
+```python
+from imrw import imread, imwrite
+
+image = imread("input.png")
+imwrite("output.png", image)
+```

imrw-0.1.0/pyproject.toml

@@ -0,0 +1,14 @@
+[build-system]
+requires = ["setuptools"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "imrw"
+version = "0.1.0"
+dependencies = [
+    "numpy",
+    "pillow",
+]
+
+[tool.setuptools.packages.find]
+where = ["src"]

imrw-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

imrw-0.1.0/src/imrw/__init__.py

@@ -0,0 +1,3 @@
+from .ops import imread, imwrite
+
+__all__ = ["imread", "imwrite"]

imrw-0.1.0/src/imrw/ops.py

@@ -0,0 +1,33 @@
+import numpy as np
+
+from typing import Any
+from pathlib import Path
+from PIL import Image
+
+
+def imread(path: str | Path) -> np.ndarray:
+    with Image.open(path) as img:
+        return np.array(img.convert("RGB"))
+
+
+def imwrite(path: str | Path, img: np.ndarray, **kwargs: Any) -> None:
+    if img.dtype != np.uint8:
+        raise ValueError(
+            f"imwrite expects a uint8 numpy array, got dtype={img.dtype}"
+        )
+
+    if img.ndim not in (2, 3):
+        raise ValueError(
+            f"imwrite expects a 2D or 3D array, got shape={img.shape}"
+        )
+
+    if img.ndim == 3 and img.shape[2] not in (1, 3, 4):
+        raise ValueError(
+            "imwrite expects channel count in {1, 3, 4}, "
+            f"got shape={img.shape}"
+        )
+
+    if img.ndim == 3 and img.shape[2] == 1:
+        img = img[:, :, 0]
+
+    Image.fromarray(img).save(path, **kwargs)

imrw-0.1.0/src/imrw.egg-info/PKG-INFO

@@ -0,0 +1,5 @@
+Metadata-Version: 2.4
+Name: imrw
+Version: 0.1.0
+Requires-Dist: numpy
+Requires-Dist: pillow

imrw-0.1.0/src/imrw.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+README.md
+pyproject.toml
+src/imrw/__init__.py
+src/imrw/ops.py
+src/imrw.egg-info/PKG-INFO
+src/imrw.egg-info/SOURCES.txt
+src/imrw.egg-info/dependency_links.txt
+src/imrw.egg-info/requires.txt
+src/imrw.egg-info/top_level.txt
+tests/test_ops.py

imrw-0.1.0/src/imrw.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

imrw-0.1.0/src/imrw.egg-info/requires.txt

@@ -0,0 +1,2 @@
+numpy
+pillow

imrw-0.1.0/src/imrw.egg-info/top_level.txt

@@ -0,0 +1 @@
+imrw

imrw-0.1.0/tests/test_ops.py

@@ -0,0 +1,65 @@
+import numpy as np
+import pytest
+from pathlib import Path
+from imrw import imread, imwrite
+
+
+def test_im_ops_roundtrip(tmp_path: Path):
+    path = tmp_path / "test.png"
+    img = np.zeros((10, 10, 3), dtype=np.uint8)
+    img[0, 0] = [255, 0, 0]
+
+    imwrite(path, img)
+    assert path.exists()
+
+    loaded = imread(path)
+    assert np.array_equal(img, loaded)
+
+
+def test_imwrite_rejects_non_uint8(tmp_path: Path):
+    path = tmp_path / "bad_dtype.png"
+    img = np.zeros((10, 10, 3), dtype=np.float32)
+
+    with pytest.raises(ValueError, match="uint8"):
+        imwrite(path, img)
+
+
+def test_imwrite_rejects_invalid_ndim(tmp_path: Path):
+    path = tmp_path / "bad_ndim.png"
+    img = np.zeros((4, 4, 3, 1), dtype=np.uint8)
+
+    with pytest.raises(ValueError, match="2D or 3D"):
+        imwrite(path, img)
+
+
+def test_imwrite_rejects_invalid_channel_count(tmp_path: Path):
+    path = tmp_path / "bad_channels.png"
+    img = np.zeros((10, 10, 2), dtype=np.uint8)
+
+    with pytest.raises(ValueError, match="channel count"):
+        imwrite(path, img)
+
+
+def test_imwrite_grayscale_2d(tmp_path: Path):
+    path = tmp_path / "gray.png"
+    img = np.full((10, 10), 128, dtype=np.uint8)
+
+    imwrite(path, img)
+    assert path.exists()
+
+
+def test_imwrite_grayscale_3d(tmp_path: Path):
+    path = tmp_path / "gray3d.png"
+    img = np.full((10, 10, 1), 128, dtype=np.uint8)
+
+    imwrite(path, img)
+    assert path.exists()
+
+
+def test_imwrite_rgba(tmp_path: Path):
+    path = tmp_path / "rgba.png"
+    img = np.zeros((10, 10, 4), dtype=np.uint8)
+    img[..., 3] = 255
+
+    imwrite(path, img)
+    assert path.exists()