Introduce warning and fix it.

astronomy-commons · Feb 15, 2024 · 1119f21 · 1119f21
1 parent a0080c6
commit 1119f21
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/hipscat/pixel_math/healpix_pixel_function.py b/src/hipscat/pixel_math/healpix_pixel_function.py
@@ -18,7 +18,7 @@ def get_pixel_argsort(pixels: List[HealpixPixel]):
     Returns:
         array of indices that sort the pixels in breadth-first order.
     """
-    if not pixels:
+    if pixels is None or len(pixels) == 0:
         return []
     # Construct a parallel list of exploded, high order pixels.
     highest_order = np.max(pixels).order

diff --git a/tests/hipscat/pixel_math/test_healpix_pixel_function.py b/tests/hipscat/pixel_math/test_healpix_pixel_function.py
@@ -14,3 +14,9 @@ def test_get_pixel_argsort(pixel_list_depth_first, pixel_list_breadth_first):
 def test_get_pixel_argsort_empty():
     argsort = get_pixel_argsort([])
     npt.assert_array_equal(argsort, [])
+
+    argsort = get_pixel_argsort(None)
+    npt.assert_array_equal(argsort, [])
+
+    argsort = get_pixel_argsort(np.array([]))
+    npt.assert_array_equal(argsort, [])