tinycwrap 0.0.0__tar.gz → 0.0.2__tar.gz

{tinycwrap-0.0.0 → tinycwrap-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tinycwrap
-Version: 0.0.0
+Version: 0.0.2
 Summary: Lightweight C-to-Python wrapper generator using CFFI and NumPy
 Author: TinyCWrap Contributors
 Requires-Python: >=3.9

{tinycwrap-0.0.0 → tinycwrap-0.0.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tinycwrap"
-version = "0.0.0"
+version = "0.0.2"
 description = "Lightweight C-to-Python wrapper generator using CFFI and NumPy"
 readme = "README.md"
 requires-python = ">=3.9"

tinycwrap-0.0.2/tests/test_examples.py

@@ -0,0 +1,97 @@
+from pathlib import Path
+
+import numpy as np
+
+import pytest
+
+from tinycwrap import CModule
+
+
+@pytest.fixture(scope="module")
+def cm():
+    c_path = Path(__file__).resolve().parent / "test_kernels.c"
+    return CModule(c_path)
+
+
+def test_dot(cm):
+    x = np.arange(10, dtype=np.float64)
+    y = np.ones_like(x)
+    assert cm.dot(x, y, len_x=len(x)) == np.sum(x * y)
+
+
+def test_scale_auto_output(cm):
+    x = np.arange(10, dtype=np.float64)
+    scaled_auto = cm.scale(x, 1.1, len_x=len(x))
+    np.testing.assert_allclose(scaled_auto, x * 1.1)
+
+
+def test_scale_explicit_output(cm):
+    x = np.arange(10, dtype=np.float64)
+    out = np.empty_like(x)
+    scaled_explicit = cm.scale(x, 2.0, len_x=len(x), out_x=out)
+    np.testing.assert_allclose(out, x * 2.0)
+    np.testing.assert_allclose(scaled_explicit, x * 2.0)
+
+
+def test_struct_wrapper(cm):
+    cp = cm.ComplexPair(real=1.5, imag=-2.5)
+    assert repr(cp) == "ComplexPair(real=1.5, imag=-2.5)"
+    assert cp.real == 1.5
+    assert cp.imag == -2.5
+    cp.real = 3.0
+    cp.imag = 4.0
+    assert cp.real == 3.0
+    assert cp.imag == 4.0
+    np.testing.assert_equal(cp.dtype.fields.keys(), {"real", "imag"})
+    assert cp.dtype["real"] == np.dtype(np.float64)
+
+
+def test_struct_argument(cm):
+    cp = cm.ComplexPair(real=3.0, imag=4.0)
+    mag_sq = cm.complex_magnitude(cp)
+    assert np.isclose(mag_sq, 5.0)
+
+
+def test_struct_array_member(cm):
+    p = cm.Particle()
+    p.pos = [1.0, 2.0, 3.0]
+    p.vel = [0.5, 0.5, 0.5]
+    np.testing.assert_allclose(p.pos, np.array([1.0, 2.0, 3.0]))
+    np.testing.assert_allclose(p.vel, np.array([0.5, 0.5, 0.5]))
+    p.pos[:] *= 2.0
+    np.testing.assert_allclose(p.pos, np.array([2.0, 4.0, 6.0]))
+
+
+def test_struct_array_argument(cm):
+    particles = cm.Particle.zeros(2)
+    particles["pos"][0] = [0.0, 0.0, 0.0]
+    particles["vel"][0] = [1.0, 0.0, 0.0]
+    particles["pos"][1] = [0.0, 0.0, 0.0]
+    particles["vel"][1] = [0.0, 2.0, 0.0]
+    ke = cm.kinetic_energy(particles, len_p=len(particles))
+    assert np.isclose(ke, 0.5 * (1.0 + 4.0))
+
+
+def test_two_output_arrays(cm):
+    data = np.array([0, 1, 2, 3, 4, 5], dtype=np.float64)
+    out1, out2 = cm.split_vectors(data)
+    np.testing.assert_allclose(out1, np.array([0, 2, 4], dtype=np.float64))
+    np.testing.assert_allclose(out2, np.array([1, 3, 5], dtype=np.float64))
+
+
+def test_merge_sorted(cm):
+    a = np.array([1.0, 2.0, 2.0, 4.0], dtype=np.float64)
+    b = np.array([2.0, 3.0], dtype=np.float64)
+    out_len = np.zeros(1, dtype=np.int32)
+    merged, out_len_val = cm.merge_sorted(a, b, out_len=out_len)
+    np.testing.assert_allclose(merged, np.array([1.0, 2.0, 3.0, 4.0]))
+    assert out_len_val == 4
+
+
+def test_struct_output_array(cm):
+    n = 4
+    particles = cm.Particle.zeros(n)
+    out_particles = cm.make_particles(3.0, out_p=particles, len_p=n)
+    np.testing.assert_array_equal(out_particles, particles)
+    np.testing.assert_allclose(particles["pos"], np.ones((n, 3)))
+    np.testing.assert_allclose(particles["vel"], np.array([[3.0, 0.0, 0.0]] * n))

tinycwrap-0.0.2/tests/test_geom.py

@@ -0,0 +1,54 @@
+from pathlib import Path
+
+import numpy as np
+
+from tinycwrap import CModule
+import pytest
+
+@pytest.fixture(scope="module")
+def cg():
+    c_path = Path(__file__).resolve().parent / "cgeom.c"
+    return CModule(c_path)
+
+
+def test_geom2d(cg):
+    pts = cg.geom2d_circle_get_points(0.0, 0.0, 1.0)
+    assert pts.shape == (101,)
+    # first point at angle 0 -> (1,0)
+    assert pts["x"][0] == 1.0
+    assert abs(pts["y"][0]) < 1e-12
+
+def test_geom2d_norm(cg):
+    n = cg.geom2d_norm(3.0, 4.0)
+    assert n == 5.0
+
+def test_circle_points_length_contract(cg):
+    pts = cg.geom2d_circle_get_points(1.0, 2.0, 3.0)
+    assert pts.shape == (101,)
+
+
+def test_return_struct_array_member(cg):
+    seg = cg.G2DSegment()
+    seg.type = 1
+    seg.data = [0.0, 0.0, 1.0, 1.0, 2.0, 2.0, 3.0, 3.0]
+    assert seg.type == 1
+    np.testing.assert_allclose(seg.data, [0.0, 0.0, 1.0, 1.0, 2.0, 2.0, 3.0, 3.0])
+
+def test_geom2d_return_strct(cg):
+    seg = cg.geom2d_line_segment_from_start_length(0.1, 0.2, 0.5, 0.4, 0.3)
+    assert seg.type == 0
+    assert seg.data[0] == 0.1
+    assert seg.data[1] == 0.2
+    assert np.isclose(seg.data[2], 0.1+0.3*0.5)
+    assert np.isclose(seg.data[3], 0.2+0.4*0.3)
+
+
+def test_geom2d_return_strct_array_member(cg):
+    seg = cg.geom2d_rectangle_to_path(1,2)
+    assert len(seg) == 4
+
+
+def test_docstring_contains_contract(cg):
+    doc = cg.geom2d_rectangle_to_path.__doc__
+    assert "Contract: len(out_segments)=4" in doc
+    assert "Auto-wrapped C function `geom2d_rectangle_to_path`." in doc

tinycwrap-0.0.2/tests/test_path.py

@@ -0,0 +1,23 @@
+from pathlib import Path
+
+import numpy as np
+import pytest
+
+from tinycwrap import CModule
+
+
+@pytest.fixture(scope="module")
+def cp():
+    return CModule(Path("tests/t1/base.c"), Path("tests/t1/path.c"))
+
+
+def test_path_get_steps_contract(cp):
+    seg = cp.G2DSegment()
+    # line from (0,0) to (1,0)
+    seg.type = 0
+    seg.data = [0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0]
+    segments = np.array([seg._data], dtype=cp.G2DSegment.dtype)
+    steps = cp.geom2d_path_get_steps(segments, ds_min=0.25)
+    expected_len = cp.geom2d_path_get_len_steps(segments, len_segments=len(segments), ds_min=0.25)
+    assert len(steps) == expected_len
+    assert np.isclose(steps[-1], cp.geom2d_path_get_length(segments, len_segments=1))

{tinycwrap-0.0.0 → tinycwrap-0.0.2}/tinycwrap/__init__.py

@@ -6,6 +6,6 @@ from .cmodule import CModule
 try:
     __version__ = version("tinycwrap")
 except PackageNotFoundError:
-    __version__ = "0.0.0"
+    __version__ = "0.0.2"
 
 __all__ = ["CModule", "__version__"]