def test_polarization_must_be_list(self) -> None:
mode: str = "Nonesense mode"
frequency_band: sar.FrequencyBand = sar.FrequencyBand.P
# Skip type hint as we are passing in an incorrect polarization.
polarizations = sar.Polarization.HV
product_type: str = "Some product"
with self.assertRaises(pystac.STACError):
SarExtension.ext(self.item).apply(
mode,
frequency_band,
polarizations, # type:ignore
product_type,
)
def test_item_ext_add_to(self) -> None:
item = pystac.Item.from_file(self.sentinel_example_uri)
item.stac_extensions.remove(SarExtension.get_schema_uri())
self.assertNotIn(SarExtension.get_schema_uri(), item.stac_extensions)
_ = SarExtension.ext(item, add_if_missing=True)
self.assertIn(SarExtension.get_schema_uri(), item.stac_extensions)
def test_extension_not_implemented(self) -> None:
# Should raise exception if Item does not include extension URI
item = pystac.Item.from_file(self.sentinel_example_uri)
item.stac_extensions.remove(SarExtension.get_schema_uri())
with self.assertRaises(pystac.ExtensionNotImplemented):
_ = SarExtension.ext(item)
# Should raise exception if owning Item does not include extension URI
asset = item.assets["measurement"]
with self.assertRaises(pystac.ExtensionNotImplemented):
_ = SarExtension.ext(asset)
# Should succeed if Asset has no owner
ownerless_asset = pystac.Asset.from_dict(asset.to_dict())
_ = SarExtension.ext(ownerless_asset)
def test_should_return_none_when_observation_direction_is_not_set(
self) -> None:
extension = SarExtension.ext(self.item)
extension.apply(
choice(ascii_letters),
choice(list(sar.FrequencyBand)),
[],
choice(ascii_letters),
)
self.assertIsNone(extension.observation_direction)
def test_required(self) -> None:
mode: str = "Nonesense mode"
frequency_band: sar.FrequencyBand = sar.FrequencyBand.P
polarizations: List[sar.Polarization] = [
sar.Polarization.HV,
sar.Polarization.VH,
]
product_type: str = "Some product"
SarExtension.ext(self.item).apply(mode, frequency_band, polarizations,
product_type)
self.assertEqual(mode, SarExtension.ext(self.item).instrument_mode)
self.assertIn(sar.INSTRUMENT_MODE_PROP, self.item.properties)
self.assertEqual(frequency_band,
SarExtension.ext(self.item).frequency_band)
self.assertIn(sar.FREQUENCY_BAND_PROP, self.item.properties)
self.assertEqual(polarizations,
SarExtension.ext(self.item).polarizations)
self.assertIn(sar.POLARIZATIONS_PROP, self.item.properties)
self.assertEqual(product_type,
SarExtension.ext(self.item).product_type)
self.assertIn(sar.PRODUCT_TYPE_PROP, self.item.properties)
self.item.validate()
def test_all(self) -> None:
mode: str = "WV"
frequency_band: sar.FrequencyBand = sar.FrequencyBand.KA
polarizations: List[sar.Polarization] = [
sar.Polarization.VV,
sar.Polarization.HH,
]
product_type: str = "Some product"
center_frequency: float = 1.2
resolution_range: float = 3.1
resolution_azimuth: float = 4.1
pixel_spacing_range: float = 5.1
pixel_spacing_azimuth: float = 6.1
looks_range: int = 7
looks_azimuth: int = 8
looks_equivalent_number: float = 9.1
observation_direction: sar.ObservationDirection = sar.ObservationDirection.LEFT
SarExtension.ext(self.item).apply(
mode,
frequency_band,
polarizations,
product_type,
center_frequency,
resolution_range,
resolution_azimuth,
pixel_spacing_range,
pixel_spacing_azimuth,
looks_range,
looks_azimuth,
looks_equivalent_number,
observation_direction,
)
self.assertEqual(center_frequency,
SarExtension.ext(self.item).center_frequency)
self.assertIn(sar.CENTER_FREQUENCY_PROP, self.item.properties)
self.assertEqual(resolution_range,
SarExtension.ext(self.item).resolution_range)
self.assertIn(sar.RESOLUTION_RANGE_PROP, self.item.properties)
self.assertEqual(resolution_azimuth,
SarExtension.ext(self.item).resolution_azimuth)
self.assertIn(sar.RESOLUTION_AZIMUTH_PROP, self.item.properties)
self.assertEqual(pixel_spacing_range,
SarExtension.ext(self.item).pixel_spacing_range)
self.assertIn(sar.PIXEL_SPACING_RANGE_PROP, self.item.properties)
self.assertEqual(pixel_spacing_azimuth,
SarExtension.ext(self.item).pixel_spacing_azimuth)
self.assertIn(sar.PIXEL_SPACING_AZIMUTH_PROP, self.item.properties)
self.assertEqual(looks_range, SarExtension.ext(self.item).looks_range)
self.assertIn(sar.LOOKS_RANGE_PROP, self.item.properties)
self.assertEqual(looks_azimuth,
SarExtension.ext(self.item).looks_azimuth)
self.assertIn(sar.LOOKS_AZIMUTH_PROP, self.item.properties)
self.assertEqual(looks_equivalent_number,
SarExtension.ext(self.item).looks_equivalent_number)
self.assertIn(sar.LOOKS_EQUIVALENT_NUMBER_PROP, self.item.properties)
self.assertEqual(observation_direction,
SarExtension.ext(self.item).observation_direction)
self.assertIn(sar.OBSERVATION_DIRECTION_PROP, self.item.properties)
self.item.validate()