def test_summaries_adds_uri(self) -> None: col = self.naip_collection col.stac_extensions = [] self.assertRaisesRegex( pystac.ExtensionNotImplemented, r"Could not find extension schema URI.*", StorageExtension.summaries, col, False, ) _ = StorageExtension.summaries(col, add_if_missing=True) self.assertIn(StorageExtension.get_schema_uri(), col.stac_extensions) StorageExtension.remove_from(col) self.assertNotIn(StorageExtension.get_schema_uri(), col.stac_extensions)
def test_tier(self) -> None: col = self.naip_collection col_dict = col.to_dict() storage_summaries = StorageExtension.summaries(col) # Get self.assertEqual(storage_summaries.tier, col_dict["summaries"]["storage:tier"]) # Set new_tier_summary = [random.choice(ascii_letters)] self.assertNotEqual(storage_summaries.tier, new_tier_summary) storage_summaries.tier = new_tier_summary self.assertEqual(storage_summaries.tier, new_tier_summary) col_dict = col.to_dict() self.assertEqual(col_dict["summaries"]["storage:tier"], new_tier_summary)
def test_platform(self) -> None: col = self.naip_collection col_dict = col.to_dict() storage_summaries = StorageExtension.summaries(col) # Get self.assertEqual(storage_summaries.platform, col_dict["summaries"]["storage:platform"]) # Set new_platform_summary = [random.choice([v for v in CloudPlatform])] self.assertNotEqual(storage_summaries.platform, new_platform_summary) storage_summaries.platform = new_platform_summary self.assertEqual(storage_summaries.platform, new_platform_summary) col_dict = col.to_dict() self.assertEqual(col_dict["summaries"]["storage:platform"], new_platform_summary)
def test_requester_pays(self) -> None: col = self.naip_collection col_dict = col.to_dict() storage_summaries = StorageExtension.summaries(col) # Get self.assertEqual( storage_summaries.requester_pays, col_dict["summaries"]["storage:requester_pays"], ) # Set new_requester_pays_summary = [True] self.assertNotEqual(storage_summaries.requester_pays, new_requester_pays_summary) storage_summaries.requester_pays = new_requester_pays_summary self.assertEqual(storage_summaries.requester_pays, new_requester_pays_summary) col_dict = col.to_dict() self.assertEqual(col_dict["summaries"]["storage:requester_pays"], new_requester_pays_summary)