Satisfy pre-commit checks

Author: jpolchlo

pystac/extensions/classification.py

@@ -24,7 +24,7 @@
 )
 from pystac.extensions.hooks import ExtensionHooks
 import pystac.extensions.item_assets as item_assets
-from pystac.extensions.raster import RasterBand, RasterExtension
+from pystac.extensions.raster import RasterBand
 from pystac.serialization.identify import STACJSONDescription, STACVersionID
 from pystac.utils import get_required, map_opt
 
@@ -614,30 +614,30 @@ def __repr__(self) -> str:
         return f"<RasterBandClassificationExtension RasterBand={self.raster_band}>"
 
 
-# class SummariesClassificationExtension(SummariesExtension):
-#     @property
-#     def classes(self) -> Optional[List[Classification]]:
-#         return map_opt(
-#             lambda classes: [Classification(c) for c in classes],
-#             self.summaries.get_list(CLASSES_PROP),
-#         )
-
-#     @classes.setter
-#     def classes(self, v: Optional[List[Classification]]) -> None:
-#         self._set_summary(CLASSES_PROP, map_opt(lambda x: [c.to_dict() for c in x], v))
-
-#     @property
-#     def bitfields(self) -> Optional[List[Bitfield]]:
-#         return map_opt(
-#             lambda bitfields: [Bitfield(b) for b in bitfields],
-#             self.summaries.get_list(BITFIELDS_PROP),
-#         )
-
-#     @bitfields.setter
-#     def bitfields(self, v: Optional[List[Bitfield]]) -> None:
-#         self._set_summary(
-#             BITFIELDS_PROP, map_opt(lambda x: [b.to_dict() for b in x], v)
-#         )
+class SummariesClassificationExtension(SummariesExtension):
+    @property
+    def classes(self) -> Optional[List[Classification]]:
+        return map_opt(
+            lambda classes: [Classification(c) for c in classes],
+            self.summaries.get_list(CLASSES_PROP),
+        )
+
+    @classes.setter
+    def classes(self, v: Optional[List[Classification]]) -> None:
+        self._set_summary(CLASSES_PROP, map_opt(lambda x: [c.to_dict() for c in x], v))
+
+    @property
+    def bitfields(self) -> Optional[List[Bitfield]]:
+        return map_opt(
+            lambda bitfields: [Bitfield(b) for b in bitfields],
+            self.summaries.get_list(BITFIELDS_PROP),
+        )
+
+    @bitfields.setter
+    def bitfields(self, v: Optional[List[Bitfield]]) -> None:
+        self._set_summary(
+            BITFIELDS_PROP, map_opt(lambda x: [b.to_dict() for b in x], v)
+        )
 
 
 class ClassificationExtensionHooks(ExtensionHooks):

tests/extensions/test_classification.py

@@ -55,17 +55,9 @@ def test_stac_extensions(landsat_item: Item) -> None:
     assert ClassificationExtension.has_extension(landsat_item)
 
 
-def test_fails_on_incompatible_object() -> None:
-    with pytest.raises(pystac.ExtensionTypeError):
-        ClassificationExtension.ext("dummy")
-
-
 def test_classification_object() -> None:
     c = Classification.create(
-        name="dummy",
-        description="empty class",
-        value=0,
-        color_hint="FF00AB"
+        name="dummy", description="empty class", value=0, color_hint="FF00AB"
     )
     assert c.name == "dummy"
     assert c.description == "empty class"
@@ -85,9 +77,9 @@ def test_bitfield_object() -> None:
             Classification.create(description="no", value=0),
             Classification.create(description="yes", value=1),
         ],
-        roles = ["data"],
-        description = "dummy description",
-        name = "dummy"
+        roles=["data"],
+        description="dummy description",
+        name="dummy",
     )
     assert b.offset == 0
     assert b.length == 1
@@ -175,14 +167,16 @@ def test_create_classes(plain_item: Item) -> None:
         Classification.create(description="yes", value=1),
     ]
     assert ext.bitfields is None
-    ext.bitfields = [Bitfield.create(
-        offset=0,
-        length=1,
-        classes=[
-            Classification.create(description="no", value=0),
-            Classification.create(description="yes", value=1),
-        ],
-    )]
+    ext.bitfields = [
+        Bitfield.create(
+            offset=0,
+            length=1,
+            classes=[
+                Classification.create(description="no", value=0),
+                Classification.create(description="yes", value=1),
+            ],
+        )
+    ]
     assert ext.classes is None