Updates to pyart unit tests to match ARM

MeteoSwiss · Aug 5, 2024 · e5efe05 · e5efe05
1 parent 54f6d16
commit e5efe05
Show file tree

Hide file tree

Showing 17 changed files with 8,771 additions and 23,650 deletions.
diff --git a/pyart/__check_build/_check_build.c b/pyart/__check_build/_check_build.c
diff --git a/pyart/correct/_fast_edge_finder.c b/pyart/correct/_fast_edge_finder.c
diff --git a/pyart/correct/_unwrap_1d.c b/pyart/correct/_unwrap_1d.c
diff --git a/pyart/correct/_unwrap_2d.c b/pyart/correct/_unwrap_2d.c
diff --git a/pyart/correct/_unwrap_3d.c b/pyart/correct/_unwrap_3d.c
diff --git a/pyart/io/_sigmetfile.c b/pyart/io/_sigmetfile.c
diff --git a/pyart/io/nexrad_interpolate.c b/pyart/io/nexrad_interpolate.c
diff --git a/pyart/io/output_to_geotiff.py b/pyart/io/output_to_geotiff.py
@@ -242,7 +242,7 @@ def _get_rgb_values(data, vmin, vmax, color_levels, cmap):
     rarr = []
     garr = []
     barr = []
-    cmap = plt.cm.get_cmap(cmap)
+    cmap = plt.get_cmap(cmap)
     for val in index:
         if not np.isnan(val):
             ind = int(np.round(val))
@@ -287,7 +287,7 @@ def _create_sld(cmap, vmin, vmax, filename, color_levels=None):
         with steps << 255 (e.g., hydrometeor ID).
 
     """
-    cmap = plt.cm.get_cmap(cmap)
+    cmap = plt.get_cmap(cmap)
     if color_levels is None:
         color_levels = 255
     name, _ = filename.split(".")

diff --git a/pyart/map/__init__.py b/pyart/map/__init__.py
@@ -5,12 +5,12 @@
 """
 
 from .gate_mapper import GateMapper  # noqa
-from .grid_mapper import map_to_grid, grid_from_radars #noqa
-from .grid_mapper import example_roi_func_constant #noqa
-from .grid_mapper import example_roi_func_dist #noqa
-from .grid_mapper import example_roi_func_dist_beam #noqa
-from .polar_to_cartesian import polar_to_cartesian #noqa
-from .polar_to_cartesian import get_earth_radius #noqa
-from .gates_to_grid import map_gates_to_grid #noqa
+from .gates_to_grid import map_gates_to_grid  # noqa
+from .grid_mapper import example_roi_func_constant  # noqa
+from .grid_mapper import example_roi_func_dist  # noqa
+from .grid_mapper import example_roi_func_dist_beam  # noqa
+from .grid_mapper import grid_from_radars  # noqa
+from .grid_mapper import map_to_grid  # noqa
+from .grid_mapper import grid_ppi_sweeps, grid_rhi_sweeps  # noqa
 
 __all__ = [s for s in dir() if not s.startswith("_")]