Merge branch 'main' into raise_filenotfounderror

Author: weiji14

.github/workflows/test-python.yml

@@ -56,19 +56,18 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       - name: Install a specific version of uv
-        uses: astral-sh/setup-uv@v3
+        uses: astral-sh/setup-uv@v6
         with:
-          enable-cache: true
-          version: "0.5.x"
+          version: "latest"
 
       - name: uv sync
         working-directory: python
-        run: uv sync --no-install-package geoindex-rs
+        run: uv sync --no-install-package async-tiff
 
       - name: maturin venv Build
         working-directory: python
         run: uv run --no-project maturin develop
 
-      # - name: Run pytest
-      #   working-directory: python
-      #   run: uv run --no-project pytest
+      - name: Run pytest
+        working-directory: python
+        run: uv run --no-project pytest --verbose

python/pyproject.toml

@@ -33,7 +33,12 @@ dev-dependencies = [
     "numpy>=1",
     "obstore>=0.5.1",
     "pip>=24.2",
-    "pytest-asyncio>=0.24.0",
+    "pytest-asyncio>=0.26.0",
     "pytest>=8.3.3",
     "ruff>=0.8.4",
 ]
+
+[tool.pytest.ini_options]
+addopts = "--color=yes"
+asyncio_default_fixture_loop_scope="function"
+asyncio_mode = "auto"

python/tests/test_cog.py

@@ -1,33 +1,30 @@
-import async_tiff
-from time import time
-
 import pytest
 
-from async_tiff import TIFF
+from async_tiff import TIFF, enums
 from async_tiff.store import LocalStore, S3Store
 
-store = S3Store("sentinel-cogs", region="us-west-2", skip_signature=True)
-path = "sentinel-s2-l2a-cogs/12/S/UF/2022/6/S2B_12SUF_20220609_0_L2A/B04.tif"
-
-tiff = await TIFF.open(path, store=store, prefetch=32768)
-
-start = time()
-tiff = await TIFF.open(path, store=store, prefetch=32768)
-end = time()
-end - start
 
-ifds = tiff.ifds
-ifd = ifds[0]
-ifd.compression
-ifd.tile_height
-ifd.tile_width
-ifd.photometric_interpretation
-gkd = ifd.geo_key_directory
-gkd.citation
-gkd.projected_type
-gkd.citation
-
-dir(gkd)
+async def test_cog_s3():
+    """
+    Ensure that TIFF.open can open a Sentinel-2 Cloud-Optimized GeoTIFF file from an
+    s3 bucket, read IFDs and GeoKeyDirectory metadata.
+    """
+    path = "sentinel-s2-l2a-cogs/12/S/UF/2022/6/S2B_12SUF_20220609_0_L2A/B04.tif"
+    store = S3Store("sentinel-cogs", region="us-west-2", skip_signature=True)
+    tiff = await TIFF.open(path=path, store=store, prefetch=32768)
+
+    ifds = tiff.ifds
+    assert len(ifds) == 5
+
+    ifd = ifds[0]
+    assert ifd.compression == enums.CompressionMethod.Deflate
+    assert ifd.tile_height == 1024
+    assert ifd.tile_width == 1024
+    assert ifd.photometric_interpretation == enums.PhotometricInterpretation.BlackIsZero
+
+    gkd = ifd.geo_key_directory
+    assert gkd.citation == "WGS 84 / UTM zone 12N"
+    assert gkd.projected_type == 32612
 
 
 async def test_cog_missing_file():