Update: tests

earthdaily · Jul 10, 2024 · 239c962 · 239c962
1 parent 21975fc
commit 239c962
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_int_geosys.py b/tests/test_int_geosys.py
@@ -160,7 +160,7 @@ def test_get_satellite_image_time_series(self):
             polygon=POLYGON
         )
         #assert dict(dataset.dims) == {'band': 7, 'y': 27, 'x': 26, 'time': 10}
-        assert all(key in dataset['dims'] for key in ['band', 'x', 'y', 'time'])
+        assert all(key in dataset for key in ['band', 'x', 'y', 'time'])
 
     def test_get_agriquest_weather_time_series(self):
         start_date = "2022-05-01"

diff --git a/tests/test_unit_geosys.py b/tests/test_unit_geosys.py
@@ -132,7 +132,7 @@ def test_get_satellite_image_time_series(self, get_response):
             polygon=POLYGON
         )
         #assert dict(dataset.dims) == {'band': 7, 'y': 13, 'x': 24, 'time': 10}
-        assert all(key in dataset['dims'] for key in ['band', 'x', 'y', 'time'])
+        assert all(key in dataset for key in ['band', 'x', 'y', 'time'])
 
 
     @patch('geosyspy.utils.http_client.HttpClient.post')

diff --git a/tests/test_unit_master_data_management_service.py b/tests/test_unit_master_data_management_service.py
@@ -110,4 +110,4 @@ def test_get_profile_fields(self, get_response):
         )
 
         response = self.service.get_profile(fields="unitProfileUnitCategories")
-        assert "unitProfileUnitCategories" in response
+        assert "id" in response