feat: first v1 test

Author: gadomski

pyproject.toml

@@ -23,7 +23,10 @@ classifiers = [
     "Programming Language :: Python :: 3.13",
 ]
 requires-python = ">=3.10"
-dependencies = ["typing-extensions>=4.12.2"]
+dependencies = [
+    "python-dateutil>=2.9.0.post0",
+    "typing-extensions>=4.12.2",
+]
 
 [project.optional-dependencies]
 validate = ["jsonschema>=4.23.0", "referencing>=0.36.2"]

src/pystac/extent.py

@@ -1,12 +1,14 @@
 from __future__ import annotations
 
+import copy
 import datetime
 import warnings
 from typing import Any, Sequence
 
 from typing_extensions import Self
 
 from .constants import DEFAULT_BBOX, DEFAULT_INTERVAL
+from .decorators import v2_deprecated
 from .errors import StacWarning
 from .types import PermissiveBbox, PermissiveInterval
 
@@ -54,7 +56,19 @@ def from_dict(cls: type[Self], d: dict[str, Any]) -> Self:
         """Creates a new spatial extent from a dictionary."""
         return cls(**d)
 
-    def __init__(self, bbox: PermissiveBbox | None = None):
+    @classmethod
+    @v2_deprecated("Use the constructor instead")
+    def from_coordinates(
+        cls: type[Self],
+        coordinates: list[Any],
+        extra_fields: dict[str, Any] | None = None,
+    ) -> Self:
+        if extra_fields:
+            return cls(coordinates, **extra_fields)
+        else:
+            return cls(coordinates)
+
+    def __init__(self, bbox: PermissiveBbox | None = None, **kwargs: Any):
         """Creates a new spatial extent."""
         self.bbox: Sequence[Sequence[float | int]]
         if bbox is None or len(bbox) == 0:
@@ -63,10 +77,13 @@ def __init__(self, bbox: PermissiveBbox | None = None):
             self.bbox = bbox  # type: ignore
         else:
             self.bbox = [bbox]  # type: ignore
+        self.extra_fields = kwargs
 
     def to_dict(self) -> dict[str, Any]:
         """Converts this spatial extent to a dictionary."""
-        return {"bbox": self.bbox}
+        d = copy.deepcopy(self.extra_fields)
+        d["bbox"] = self.bbox
+        return d
 
 
 class TemporalExtent:
@@ -77,6 +94,11 @@ def from_dict(cls: type[Self], d: dict[str, Any]) -> Self:
         """Creates a new temporal extent from a dictionary."""
         return cls(**d)
 
+    @classmethod
+    def from_now(cls: type[Self]) -> Self:
+        """Creates a new temporal extent that starts now and has no end time."""
+        return cls([[datetime.datetime.now(tz=datetime.timezone.utc), None]])
+
     def __init__(
         self,
         interval: PermissiveInterval | None = None,

src/pystac/stac_object.py

@@ -21,6 +21,7 @@
 from .link import Link
 
 if TYPE_CHECKING:
+    from .catalog import Catalog
     from .io import Read, Write
 
 
@@ -81,6 +82,9 @@ def from_dict(
         d: dict[str, Any],
         *,
         href: str | None = None,
+        root: Catalog | None = None,  # TODO deprecation warning
+        migrate: bool = False,
+        preserve_dict: bool = True,  # TODO deprecation warning
         reader: Read | None = None,
         writer: Write | None = None,
     ) -> STACObject:

tests/test_extent.py

@@ -3,7 +3,7 @@
 
 import pytest
 
-from pystac import StacWarning, TemporalExtent
+from pystac import SpatialExtent, StacWarning, TemporalExtent
 
 
 def test_temporal_with_datetimes() -> None:
@@ -45,3 +45,14 @@ def test_temporal_with_bad_tail() -> None:
         )
     d = extent.to_dict()
     assert d == {"interval": [["2025-02-11T00:00:00Z", None]]}
+
+
+def test_temporal_from_now() -> None:
+    extent = TemporalExtent.from_now()
+    assert isinstance(extent.interval[0][0], str)
+    assert extent.interval[0][1] is None
+
+
+def test_spatial_from_coordinates() -> None:
+    with pytest.warns(FutureWarning):
+        SpatialExtent.from_coordinates([-180, -90, 180, 90])

tests/test_item.py

@@ -37,3 +37,10 @@ def test_warn_include_self_link() -> None:
 def test_warn_transform_hrefs() -> None:
     with pytest.warns(FutureWarning):
         Item("an-id").to_dict(transform_hrefs=True)
+
+
+def test_from_dict_migrate() -> None:
+    d = Item("an-id").to_dict()
+    d["stac_version"] = "1.0.0"
+    item = Item.from_dict(d, migrate=True)
+    item.stac_version == "1.1.0"

tests/v1/__init__.py

@@ -0,0 +1,14 @@
+import json
+from typing import Any
+
+import pytest
+
+from .utils import TestCases
+
+
+@pytest.fixture
+def sample_item_dict() -> dict[str, Any]:
+    m = TestCases.get_path("data-files/item/sample-item.json")
+    with open(m) as f:
+        item_dict: dict[str, Any] = json.load(f)
+    return item_dict

tests/v1/conftest.py

@@ -0,0 +1,81 @@
+{
+    "type": "Feature",
+    "stac_version": "1.1.0",
+    "id": "CS3-20160503_132131_05",
+    "properties": {
+        "datetime": "2016-05-03T13:22:30.040000Z",
+        "title": "A CS3 item",
+        "license": "PDDL-1.0",
+        "providers": [
+            {
+                "name": "CoolSat",
+                "roles": [
+                    "producer",
+                    "licensor"
+                ],
+                "url": "https://cool-sat.com/"
+            }
+        ]
+    },
+    "geometry": {
+        "type": "Polygon",
+        "coordinates": [
+            [
+                [
+                    -122.308150179,
+                    37.488035566
+                ],
+                [
+                    -122.597502109,
+                    37.538869539
+                ],
+                [
+                    -122.576687533,
+                    37.613537207
+                ],
+                [
+                    -122.2880486,
+                    37.562818007
+                ],
+                [
+                    -122.308150179,
+                    37.488035566
+                ]
+            ]
+        ]
+    },
+    "links": [
+        {
+            "rel": "collection",
+            "href": "https://raw.githubusercontent.com/radiantearth/stac-spec/v0.8.1/collection-spec/examples/sentinel2.json"
+        }
+    ],
+    "assets": {
+        "analytic": {
+            "href": "http://cool-sat.com/catalog/CS3-20160503_132130_04/analytic.tif",
+            "title": "4-Band Analytic",
+            "product": "http://cool-sat.com/catalog/products/analytic.json",
+            "type": "image/tiff; application=geotiff; profile=cloud-optimized",
+            "roles": [
+                "data",
+                "analytic"
+            ]
+        },
+        "thumbnail": {
+            "href": "http://cool-sat.com/catalog/CS3-20160503_132130_04/thumbnail.png",
+            "title": "Thumbnail",
+            "type": "image/png",
+            "roles": [
+                "thumbnail"
+            ]
+        }
+    },
+    "bbox": [
+        -122.59750209,
+        37.48803556,
+        -122.2880486,
+        37.613537207
+    ],
+    "stac_extensions": [],
+    "collection": "CS3"
+}

tests/v1/data-files/item/sample-item.json

@@ -0,0 +1,28 @@
+from copy import deepcopy
+from typing import Any
+
+from pystac import Item
+
+from . import utils
+
+
+def test_to_from_dict(sample_item_dict: dict[str, Any]) -> None:
+    param_dict = deepcopy(sample_item_dict)
+
+    utils.assert_to_from_dict(Item, param_dict)
+    item = Item.from_dict(param_dict)
+    assert item.id == "CS3-20160503_132131_05"
+
+    # test asset creation additional field(s)
+    assert (
+        item.assets["analytic"].extra_fields["product"]
+        == "http://cool-sat.com/catalog/products/analytic.json"
+    )
+    assert len(item.assets["thumbnail"].extra_fields) == 0
+
+    # test that the parameter is preserved
+    assert param_dict == sample_item_dict
+
+    # assert that the parameter is preserved regardless of preserve_dict
+    Item.from_dict(param_dict, preserve_dict=False)
+    assert param_dict == sample_item_dict