tests, cleanup: Updated tests with more ranges. Cleaned up misc comme…

…nted out print statements
NOAA-HES-Capstone · Mar 17, 2024 · f78c34a · f78c34a
1 parent b8f44ea
commit f78c34a
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 10 deletions.
diff --git a/icedyno/preprocess/geolocation.py b/icedyno/preprocess/geolocation.py
@@ -70,9 +70,7 @@ def polar_xy_to_lonlat(
     lat = lat * 180 / np.pi
     lon = np.arctan2(x, -y)
     lon = lon * 180 / np.pi
-    print("this is lon: ", lon)
-    # lon = lon + np.less(lon, 0) * 360
-    # print("again lon: ", lon)
+
     return lon, lat
 
 

diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/test_geolocation.py b/tests/test_geolocation.py
@@ -5,12 +5,15 @@
 
 def test_xy_to_latlon_inversion():
     """Test that xy to latlon and latlon to xy invert each other"""
-    lat = 77.508333
-    lon = -164.754167
+    lats = [77.508333, 80, 67, 55]
+    lons = [-164.754167, -80, 90, 155]
 
-    xy_converted_coordinates = geolocation.polar_lonlat_to_xy(
-        longitude=lon, latitude=lat
-    )
+    for i in range(len(lats)):
+        lat = lats[i]
+        lon = lons[i]
+        xy_converted_coordinates = geolocation.polar_lonlat_to_xy(
+            longitude=lon, latitude=lat
+        )
 
-    reverted_lat_lon = geolocation.polar_xy_to_lonlat(*xy_converted_coordinates)
-    assert np.allclose((lon, lat), reverted_lat_lon)
+        reverted_lat_lon = geolocation.polar_xy_to_lonlat(*xy_converted_coordinates)
+        assert np.allclose((lon, lat), reverted_lat_lon)