Skip to content

Commit

Permalink
Create new catalog for each SAT extension test
Browse files Browse the repository at this point in the history
  • Loading branch information
duckontheweb committed Jul 2, 2021
1 parent 648aafd commit eed30aa
Showing 1 changed file with 18 additions and 11 deletions.
29 changes: 18 additions & 11 deletions tests/extensions/test_sat.py
Original file line number Diff line number Diff line change
Expand Up @@ -223,12 +223,15 @@ class SatSummariesTest(unittest.TestCase):
def setUp(self) -> None:
self.maxDiff = None

self.collection = pystac.Collection.from_file(
@staticmethod
def collection() -> pystac.Collection:
return pystac.Collection.from_file(
TestCases.get_path("data-files/collections/multi-extent.json")
)

def test_platform_international_designation(self) -> None:
summaries_ext = SatExtension.summaries(self.collection)
collection = self.collection()
summaries_ext = SatExtension.summaries(collection)
platform_international_designator_list = ["2018-080A"]

summaries_ext.platform_international_designator = ["2018-080A"]
Expand All @@ -238,15 +241,16 @@ def test_platform_international_designation(self) -> None:
platform_international_designator_list,
)

summaries_dict = self.collection.to_dict()["summaries"]
summaries_dict = collection.to_dict()["summaries"]

self.assertEqual(
summaries_dict["sat:platform_international_designator"],
platform_international_designator_list,
)

def test_orbit_state(self) -> None:
summaries_ext = SatExtension.summaries(self.collection)
collection = self.collection()
summaries_ext = SatExtension.summaries(collection)
orbit_state_list = [OrbitState.ASCENDING]

summaries_ext.orbit_state = orbit_state_list
Expand All @@ -256,15 +260,16 @@ def test_orbit_state(self) -> None:
orbit_state_list,
)

summaries_dict = self.collection.to_dict()["summaries"]
summaries_dict = collection.to_dict()["summaries"]

self.assertEqual(
summaries_dict["sat:orbit_state"],
orbit_state_list,
)

def test_absolute_orbit(self) -> None:
summaries_ext = SatExtension.summaries(self.collection)
collection = self.collection()
summaries_ext = SatExtension.summaries(collection)
absolute_orbit_range = RangeSummary(2000, 3000)

summaries_ext.absolute_orbit = absolute_orbit_range
Expand All @@ -274,15 +279,16 @@ def test_absolute_orbit(self) -> None:
absolute_orbit_range,
)

summaries_dict = self.collection.to_dict()["summaries"]
summaries_dict = collection.to_dict()["summaries"]

self.assertEqual(
summaries_dict["sat:absolute_orbit"],
absolute_orbit_range.to_dict(),
)

def test_relative_orbit(self) -> None:
summaries_ext = SatExtension.summaries(self.collection)
collection = self.collection()
summaries_ext = SatExtension.summaries(collection)
relative_orbit_range = RangeSummary(50, 100)

summaries_ext.relative_orbit = relative_orbit_range
Expand All @@ -292,15 +298,16 @@ def test_relative_orbit(self) -> None:
relative_orbit_range,
)

summaries_dict = self.collection.to_dict()["summaries"]
summaries_dict = collection.to_dict()["summaries"]

self.assertEqual(
summaries_dict["sat:relative_orbit"],
relative_orbit_range.to_dict(),
)

def test_anx_datetime(self) -> None:
summaries_ext = SatExtension.summaries(self.collection)
collection = self.collection()
summaries_ext = SatExtension.summaries(collection)
anx_datetime_range = RangeSummary(
str_to_datetime("2020-01-01T00:00:00.000Z"),
str_to_datetime("2020-01-02T00:00:00.000Z"),
Expand All @@ -313,7 +320,7 @@ def test_anx_datetime(self) -> None:
anx_datetime_range,
)

summaries_dict = self.collection.to_dict()["summaries"]
summaries_dict = collection.to_dict()["summaries"]

self.assertDictEqual(
summaries_dict["sat:anx_datetime"],
Expand Down

0 comments on commit eed30aa

Please sign in to comment.