Update algorithms.py

Allow all defined numpy complex dtypes to work as well as complex128.  Note that by default the test suite only tests the complex128 case, but users can add or alter that default by modifying the test suite source code to test other cases.

Signed-off-by: Michael Tiemann <[email protected]>

Author: MichaelTiemannOSC

pandas/core/algorithms.py

@@ -172,8 +172,9 @@ def _ensure_data(values: ArrayLike) -> np.ndarray:
     elif is_complex_dtype(values.dtype):
         # error: Item "ExtensionDtype" of "Union[Any, ExtensionDtype]"
         # has no attribute "itemsize"
-        if values.dtype.itemsize == 16:  # type: ignore[union-attr]
-            # We have support for complex128
+        if values.dtype.itemsize in [32, 24, 16, 8]:  # type: ignore[union-attr]
+            # The test suite tests support for complex128; we presume that
+            # complex64, complex192, and complex256 work as well
             return np.asarray(values)
         return cast(np.ndarray, values)