Support other list-like inputs for plot_healpix_map (#606)

Author: delucchi-cmu

src/hats/inspection/_plotting.py

@@ -397,7 +397,7 @@ def _initialize_wcs_axes(
     return fig, ax, wcs
 
 
-# pylint: disable=import-outside-toplevel,import-error,disable=too-many-locals
+# pylint: disable=import-outside-toplevel,import-error,too-many-locals
 def _plot_healpix_value_map(ipix, depth, values, ax, wcs, cmap="viridis", norm=None, cbar=True, **kwargs):
     """Perform the plotting of a healpix pixel map."""
     import matplotlib.pyplot as plt
@@ -406,6 +406,8 @@ def _plot_healpix_value_map(ipix, depth, values, ax, wcs, cmap="viridis", norm=N
 
     # create dict mapping depth to ipix and values
     depth_ipix_d = {}
+    values = np.array(values)
+    ipix = np.array(ipix)
 
     for d in np.unique(depth):
         mask = depth == d

tests/hats/inspection/test_visualize_catalog.py

@@ -2,6 +2,7 @@
 
 import astropy.units as u
 import numpy as np
+import pandas as pd
 import pytest
 from astropy.coordinates import Angle, SkyCoord
 from mocpy import MOC, WCS
@@ -843,6 +844,37 @@ def test_plot_kwargs():
     np.testing.assert_array_equal(col.get_array(), pix_map)
 
 
+def test_plot_healpix_map_types():
+    """Pass healpix map, using various list-like types."""
+    pytest.importorskip("matplotlib.pyplot")
+
+    # First, use all numpy arrays to get the golden value.
+    length = 192
+    pix_map_np = np.arange(length)
+    _, ax = plot_healpix_map(pix_map_np)
+    num_np_paths = len(ax.collections[0].get_paths())
+
+    pix_map = list(np.arange(length))
+    _, ax = plot_healpix_map(pix_map)
+    assert len(ax.collections) > 0
+    assert len(ax.collections[0].get_paths()) == num_np_paths
+
+    order = 2
+    ipix = list(np.arange(length))
+    pix_map = list(range(0, length))
+    depth = list(np.full(length, fill_value=order))
+    _, ax = plot_healpix_map(pix_map, ipix=ipix, depth=depth)
+    assert len(ax.collections) > 0
+    assert len(ax.collections[0].get_paths()) == num_np_paths
+
+    ipix = pd.Series(range(0, length))
+    pix_map = range(0, length)
+    depth = pd.Series(np.full(length, fill_value=order))
+    _, ax = plot_healpix_map(pix_map, ipix=ipix, depth=depth)
+    assert len(ax.collections) > 0
+    assert len(ax.collections[0].get_paths()) == num_np_paths
+
+
 def test_plot_existing_fig():
     plt = pytest.importorskip("matplotlib.pyplot")