healpix-geo 0.0.5__cp310-cp310-win_amd64.whl → 0.0.6__cp310-cp310-win_amd64.whl

healpix_geo/__init__.py

@@ -1,3 +1,3 @@
-from healpix_geo import nested, ring
+from healpix_geo import nested, ring, slices
 
-__all__ = ["nested", "ring"]
+__all__ = ["nested", "ring", "slices"]

healpix_geo/slices.py

@@ -0,0 +1,4 @@
+from healpix_geo.healpix_geo import slices as slices_
+
+Slice = slices_.Slice  # noqa: F401
+ConcreteSlice = slices_.ConcreteSlice  # noqa: F401

healpix_geo/tests/test_index.py

@@ -121,6 +121,95 @@ class TestRangeMOCIndex:
 
         np.testing.assert_equal(actual.cell_ids(), expected)
 
+    @pytest.mark.parametrize(
+        ["level", "cell_ids", "indexer"],
+        (
+            pytest.param(
+                0,
+                np.arange(12, dtype="uint64"),
+                slice(None),
+                id="base cells-slice-full",
+            ),
+            pytest.param(
+                0,
+                np.arange(12, dtype="uint64"),
+                slice(None, 6),
+                id="base cells-slice-left_open",
+            ),
+            pytest.param(
+                0,
+                np.arange(12, dtype="uint64"),
+                slice(2, None),
+                id="base cells-slice-right_open",
+            ),
+            pytest.param(
+                0,
+                np.arange(12, dtype="uint64"),
+                slice(2, 7),
+                id="base cells-slice-domain",
+            ),
+            pytest.param(
+                0,
+                np.arange(12, dtype="uint64"),
+                np.arange(12, dtype="uint64"),
+                id="base cells-array-full",
+            ),
+            pytest.param(
+                0,
+                np.arange(12, dtype="uint64"),
+                np.arange(2, 7, dtype="uint64"),
+                id="base cells-array-domain",
+            ),
+            pytest.param(
+                0,
+                np.arange(12, dtype="uint64"),
+                np.array([1, 2, 3, 7, 8, 9, 10], dtype="uint64"),
+                id="base cells-array-disconnected",
+            ),
+            pytest.param(
+                3,
+                np.arange(12 * 4**3, dtype="uint64"),
+                slice(None, 15),
+                id="level 3 cells-slice-left_open",
+            ),
+            pytest.param(
+                1,
+                np.array([0, 1, 2, 4, 5, 11, 12, 13, 25, 26, 27], dtype="uint64"),
+                np.array([2, 5, 11, 12, 25, 27], dtype="uint64"),
+                id="list of level 1 cells-array-disconnected",
+            ),
+            pytest.param(
+                4,
+                np.arange(1 * 4**4, 2 * 4**4, dtype="uint64"),
+                slice(260, 280),
+                id="single level 4 base cell-slice-domain",
+            ),
+        ),
+    )
+    def test_sel(self, level, cell_ids, indexer):
+        if isinstance(indexer, slice):
+            n = slice(
+                indexer.start if indexer.start is not None else 0,
+                (
+                    indexer.stop + 1
+                    if indexer.stop is not None
+                    else int(np.max(cell_ids)) + 1
+                ),
+                indexer.step if indexer.step is not None else 1,
+            )
+            range_ = np.arange(n.start, n.stop, n.step, dtype="uint64")
+            condition = np.isin(cell_ids, range_)
+        else:
+            condition = np.isin(cell_ids, indexer)
+        expected_cell_ids = cell_ids[condition]
+
+        index = healpix_geo.nested.RangeMOCIndex.from_cell_ids(level, cell_ids)
+
+        actual_indexer, actual_moc = index.sel(indexer)
+
+        np.testing.assert_equal(cell_ids[actual_indexer], expected_cell_ids)
+        np.testing.assert_equal(actual_moc.cell_ids(), expected_cell_ids)
+
     @pytest.mark.parametrize(
         ["depth", "cell_ids"],
         (

healpix_geo/tests/test_slices.py

@@ -0,0 +1,116 @@
+import pytest
+
+from healpix_geo import slices
+
+missing = object()
+
+
+class TestSlice:
+    @pytest.mark.parametrize("start", [0, None, 10])
+    @pytest.mark.parametrize("stop", [3, None, 10])
+    @pytest.mark.parametrize("step", [1, None, 2, missing])
+    def test_init(self, start, stop, step):
+        if step is missing:
+            vals = (start, stop)
+            step = None
+        else:
+            vals = (start, stop, step)
+
+        actual = slices.Slice(*vals)
+
+        assert actual.start == start
+        assert actual.stop == stop
+        assert actual.step == step
+
+    @pytest.mark.parametrize("start", [0, None, 10])
+    @pytest.mark.parametrize("stop", [3, None, 10])
+    @pytest.mark.parametrize("step", [1, None, 2, -1])
+    def test_repr(self, start, stop, step):
+        slice_ = slices.Slice(start, stop, step)
+        actual = repr(slice_)
+
+        expected = repr(slice(start, stop, step)).title()
+
+        assert actual == expected
+
+    @pytest.mark.parametrize(
+        "pyslice",
+        (
+            slice(None),
+            slice(None, 2),
+            slice(3, None),
+            slice(None, None, -1),
+            slice(10, None, -1),
+            slice(3, -2),
+        ),
+    )
+    def test_from_pyslice(self, pyslice):
+        actual = slices.Slice.from_pyslice(pyslice)
+
+        assert actual.start == pyslice.start
+        assert actual.start == pyslice.start
+        assert actual.start == pyslice.start
+
+    @pytest.mark.parametrize(
+        "pyslice",
+        (
+            slice(None),
+            slice(None, 2),
+            slice(3, None),
+            slice(None, None, -1),
+            slice(10, None, -1),
+            slice(3, -2),
+        ),
+    )
+    @pytest.mark.parametrize("size", (3, 8, 13))
+    def test_roundtrip_pyslice(self, pyslice, size):
+        slice_ = slices.Slice.from_pyslice(pyslice)
+
+        actual = slice_.as_pyslice()
+        assert actual == pyslice
+
+    @pytest.mark.parametrize(
+        "pyslice",
+        (
+            slice(None),
+            slice(None, 2),
+            slice(3, None),
+            slice(None, None, -1),
+            slice(10, None, -1),
+            slice(3, -2),
+        ),
+    )
+    @pytest.mark.parametrize("size", (3, 8, 13))
+    def test_as_concrete(self, pyslice, size):
+        slice_ = slices.Slice(pyslice.start, pyslice.stop, pyslice.step)
+
+        actual = slice_.as_concrete(size)
+        expected = slice(*pyslice.indices(size))
+
+        assert actual.start == expected.start
+        assert actual.stop == expected.stop
+        assert actual.step == expected.step
+
+    def test_in_dict(self):
+        slice1 = slices.Slice(0, 4)
+        slice2 = slices.Slice(4, 6)
+
+        d = {slice1: 1, slice2: 2}
+
+        assert d[slice1] == 1
+        assert d[slice2] == 2
+
+    @pytest.mark.parametrize(
+        ["vals", "expected"],
+        (
+            ((0, 4), True),
+            ((1, 4), False),
+            ((0, 4, 1), False),
+        ),
+    )
+    def test_compare(self, vals, expected):
+        slice_ = slices.Slice(0, 4)
+        other = slices.Slice(*vals)
+
+        actual = slice_ == other
+        assert actual == expected

{healpix_geo-0.0.5.dist-info → healpix_geo-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: healpix-geo
-Version: 0.0.5
+Version: 0.0.6
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
@@ -17,11 +17,5 @@ Requires-Dist: tabulate ; extra == 'docs'
 Provides-Extra: tests
 Provides-Extra: docs
 License-File: LICENSE
-License: Apache-2.0
+License-Expression: Apache-2.0
 Requires-Python: >=3.10
-Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
-
-# `healpix-geo`
-
-This package integrates with `cds-healpix-rust` and `cds-healpix-python` to support specialized algorithms for the geosciences.
-

healpix_geo-0.0.6.dist-info/RECORD

@@ -0,0 +1,16 @@
+healpix_geo-0.0.6.dist-info/METADATA,sha256=oF5iTjtlZpD2GwZkueXnW3xfhZ1TBhmZoPiudqEgCoQ,783
+healpix_geo-0.0.6.dist-info/WHEEL,sha256=gTksDcUm4vEEt-WSZ1CnYcSbBNyUtm2mp9lzlf3Yd0w,96
+healpix_geo-0.0.6.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+healpix_geo/__init__.py,sha256=KemntjpEc5RlTjkX4SWMBT8f0afpSSowuBGHAIwFOHE,88
+healpix_geo/healpix_geo.cp310-win_amd64.pyd,sha256=xKzIFGC66RH-vEwv4hdVuRwgn1GbLBBpbhfZOXSdSUI,1240576
+healpix_geo/nested.py,sha256=mQ75zRvY8gI20nkS9ayruY6VgoKoRyHzuoLuCu_rafw,12983
+healpix_geo/ring.py,sha256=G6zGJDxfAfWH3pOJacCoV23TewnhfImkO1HWi6X3CdM,10601
+healpix_geo/slices.py,sha256=SdErlNhiGF1B_SCJEcwd0pBIt3yegyUk1aoDK8UVT_M,147
+healpix_geo/tests/test_distances.py,sha256=_oxekYaotPdb2M2NqI4pVyFOWiCXqBNqJnDmLf-j3jM,2583
+healpix_geo/tests/test_index.py,sha256=TsW1-tiSVZjUAISyTc42-rpIrCwDIzFz6a9RYaV0hck,8061
+healpix_geo/tests/test_inference.py,sha256=wmpi-5BtLuk8MGtlU4b88Mwo5Q2WbMPLVPyf1mwCxcY,7367
+healpix_geo/tests/test_neighbours.py,sha256=nfBIIwssCZWeSQgI2loJMHYLLm_iAKiQ9X_eXyuzgg4,1227
+healpix_geo/tests/test_slices.py,sha256=F6j0bHR-iMly4WMqNVy9D9MosaZw67lBX_mK0hCOkpo,3238
+healpix_geo/tests/test_zoom.py,sha256=fNiNEwwB5KlcxWGFZEfu76f_nZlyb5esVjw5JO10qY4,2339
+healpix_geo/utils.py,sha256=1N4VkjkhJm5SN1vCyDCLuBC8L6gOGszCEy1cUmdVjoc,746
+healpix_geo-0.0.6.dist-info/RECORD,,

{healpix_geo-0.0.5.dist-info → healpix_geo-0.0.6.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: maturin (1.9.0)
+Generator: maturin (1.9.3)
 Root-Is-Purelib: false
 Tag: cp310-cp310-win_amd64

healpix_geo-0.0.5.dist-info/RECORD

@@ -1,14 +0,0 @@
-healpix_geo-0.0.5.dist-info/METADATA,sha256=QoTnNnL05PQ7dWy7y24dSOPssmBrC1AyWFKbaAzFMQ0,990
-healpix_geo-0.0.5.dist-info/WHEEL,sha256=oIYcReGXK7h7Rxp85Q3K9Z3I1xK99xgU5My5R5-0sSU,96
-healpix_geo-0.0.5.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-healpix_geo/__init__.py,sha256=JJc1U_jRprkKj-0zzlfHex7rHAAILAF0bkeQ--D15WU,70
-healpix_geo/healpix_geo.cp310-win_amd64.pyd,sha256=HEYoLGXDN8p3jH5h1mL1-Wc9B4YEqzYytxsIWBaKJ0I,1127936
-healpix_geo/nested.py,sha256=mQ75zRvY8gI20nkS9ayruY6VgoKoRyHzuoLuCu_rafw,12983
-healpix_geo/ring.py,sha256=G6zGJDxfAfWH3pOJacCoV23TewnhfImkO1HWi6X3CdM,10601
-healpix_geo/tests/test_distances.py,sha256=_oxekYaotPdb2M2NqI4pVyFOWiCXqBNqJnDmLf-j3jM,2583
-healpix_geo/tests/test_index.py,sha256=Xwa6gWHYHD-vnqxOOf2fmLic1nyKZ1TSry4JQLwwYH8,4865
-healpix_geo/tests/test_inference.py,sha256=wmpi-5BtLuk8MGtlU4b88Mwo5Q2WbMPLVPyf1mwCxcY,7367
-healpix_geo/tests/test_neighbours.py,sha256=nfBIIwssCZWeSQgI2loJMHYLLm_iAKiQ9X_eXyuzgg4,1227
-healpix_geo/tests/test_zoom.py,sha256=fNiNEwwB5KlcxWGFZEfu76f_nZlyb5esVjw5JO10qY4,2339
-healpix_geo/utils.py,sha256=1N4VkjkhJm5SN1vCyDCLuBC8L6gOGszCEy1cUmdVjoc,746
-healpix_geo-0.0.5.dist-info/RECORD,,