Tests, Github Actions, and Python 3.8, 3.9 support (#4)

Adds a fixture, test, and the github workflow required to run the tests
on github actions.

I originally added a test data file copied from
https://github.com/stac-utils/pystac/blob/v1.6.1/tests/data-files/examples/1.0.0/simple-item.json
but then I realized I can just link out to that file where it lives.

I still need to decide whether tests should do reads from remote files
or not.

Author: jsignell

.github/workflows/tests.yml

@@ -0,0 +1,50 @@
+name: Tests
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+  workflow_dispatch:
+  schedule:
+    - cron: "20 22 * * 0"
+
+concurrency:
+  group: ${{ github.ref }}
+  cancel-in-progress: true
+
+defaults:
+  run:
+    shell: bash -l {0}
+
+jobs:
+  test:
+    name: ${{ matrix.python-version }}
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest]
+        python-version: ["3.8", "3.9", "3.10", "3.11"]
+        environment-file: ["environment.yaml"]
+    defaults:
+      run:
+        shell: bash -l {0}
+
+    steps:
+      - name: checkout repo
+        uses: actions/[email protected]
+
+      - name: setup micromamba
+        uses: mamba-org/provision-with-micromamba@main
+        with:
+          environment-file: ${{ matrix.environment-file }}
+          micromamba-version: "latest"
+          extra-specs: python=${{ matrix.python-version }}
+          channel-priority: "flexible"
+
+      - name: install xpystac
+        run: pip install .
+
+      - name: run tests
+        id: status
+        run: pytest -v .

.pre-commit-config.yaml

@@ -1,10 +1,25 @@
 
 repos:
+-   repo: https://github.com/MarcoGorelli/absolufy-imports
+    rev: v0.3.1
+    hooks:
+      - id: absolufy-imports
+        name: absolufy-imports
+-   repo: https://github.com/pycqa/isort
+    rev: 5.12.0
+    hooks:
+      - id: isort
+        language_version: python3
+-   repo: https://github.com/asottile/pyupgrade
+    rev: v3.2.2
+    hooks:
+      - id: pyupgrade
+        args:
+          - --py38-plus
 -   repo: https://github.com/psf/black
-    rev: "22.12.0"
+    rev: "23.1.0"
     hooks:
     -   id: black
-        additional_dependencies: ['click==8.0.4']
 -   repo: https://github.com/pycqa/flake8
     rev: 6.0.0
     hooks:

environment.yaml

@@ -0,0 +1,19 @@
+name: xpystac-dev
+channels:
+  - conda-forge
+  - nodefaults
+dependencies:
+  - python
+  # required
+  - pystac
+  - xarray
+  # optional
+  - fsspec
+  - planetary-computer
+  - pystac-client
+  - requests
+  - rioxarray
+  - stackstac
+  - zarr
+  # testing
+  - pytest

tests/__init__.py

@@ -0,0 +1,8 @@
+import pystac
+import pytest
+
+
+@pytest.fixture(scope="module")
+def simple_item() -> pystac.Item:
+    path = "https://raw.githubusercontent.com/stac-utils/pystac/2.0/tests/data-files/examples/1.0.0/simple-item.json"
+    return pystac.Item.from_file(path)

tests/conftest.py

@@ -0,0 +1,10 @@
+import pystac
+
+from xpystac.core import to_xarray
+
+
+def test_asset_to_xarray(simple_item):
+    asset = simple_item.assets["visual"]
+    assert asset.media_type == pystac.MediaType.COG
+    ds = to_xarray(asset)
+    assert ds

tests/test_core.py

@@ -1,9 +1,10 @@
 import functools
+from typing import List, Union
 
 import pystac
 import xarray
 
-from .utils import _import_optional_dependency
+from xpystac.utils import _import_optional_dependency
 
 
 @functools.singledispatch
@@ -18,8 +19,8 @@ def to_xarray(item, **kwargs) -> xarray.Dataset:
 @to_xarray.register(pystac.Item)
 @to_xarray.register(pystac.ItemCollection)
 def _(
-    obj: pystac.Item | pystac.ItemCollection,
-    drop_variables: str | list[str] = None,
+    obj: Union[pystac.Item, pystac.ItemCollection],
+    drop_variables: Union[str, List[str]] = None,
     **kwargs,
 ) -> xarray.Dataset:
     stackstac = _import_optional_dependency("stackstac")

xpystac/core.py

@@ -1,15 +1,17 @@
+from typing import List, Union
+
 import pystac
 from xarray.backends import BackendEntrypoint
 
-from .core import to_xarray
+from xpystac.core import to_xarray
 
 
 class STACBackend(BackendEntrypoint):
     def open_dataset(
         self,
         obj,
         *,
-        drop_variables: str | list[str] = None,
+        drop_variables: Union[str, List[str]] = None,
         **kwargs,
     ):
         return to_xarray(obj, drop_variables=drop_variables, **kwargs)