FIX: Add unitless comparison operators to Quantity class

greglucas · greglucas · commit f716ea3ddc35 · 2022-07-30T08:56:03.000-06:00
diff --git a/lib/matplotlib/tests/test_image.py b/lib/matplotlib/tests/test_image.py
@@ -1183,6 +1183,11 @@ def __array_ufunc__(self, ufunc, method, *inputs, **kwargs):
                 unit = f"{u0}*{u1}"
             elif ufunc == np.divide:
                 unit = f"{u0}/({u1})"
+            elif ufunc in (np.greater, np.greater_equal,
+                           np.equal, np.not_equal,
+                           np.less, np.less_equal):
+                # Comparisons produce unitless booleans for output
+                unit = None
             else:
                 return NotImplemented
             out_arr = func(i0.view(np.ndarray), i1.view(np.ndarray), **kwargs)
