complex 64 and 128 testing

Author: alimcmaster1

pandas/tests/reductions/test_reductions.py

@@ -1487,3 +1487,33 @@ def test_mode_boolean_with_na(self):
         result = ser.mode()
         expected = Series({0: True}, dtype="boolean")
         tm.assert_series_equal(result, expected)
+
+    @pytest.mark.parametrize(
+        "array,expected,dtype",
+        [
+            ([0, 1j, 1, 1, 1 + 1j, 1 + 2j], Series([1], dtype=np.complex128), np.complex128),
+            ([0, 1j, 1, 1, 1 + 1j, 1 + 2j], Series([1], dtype=np.complex64), np.complex64),
+            ([1 + 1j, 2j, 1 + 1j], Series([1 + 1j], dtype=np.complex128), np.complex128),
+        ],
+    )
+    def test_unimode_complex(self, array, expected, dtype):
+        result = Series(array, dtype=dtype).mode()
+        tm.assert_series_equal(result, expected)
+
+    @pytest.mark.parametrize(
+        "array,expected,dtype",
+        [
+            (
+                # no modes
+                [0, 1j, 1, 1 + 1j, 1 + 2j],
+                Series([0j, 1j, 1 + 0j, 1 + 1j, 1 + 2j], dtype=np.complex128),
+                np.complex128
+            ),
+            ([1 + 1j, 2j, 1 + 1j, 2j, 3], Series([2j, 1 + 1j], dtype=np.complex64), np.complex64),
+        ],
+    )
+    def test_multimode_complex(self, array, expected,dtype):
+        # mode tries to sort multimodal series.
+        # Complex numbers are sorted by their magnitude
+        result = Series(array, dtype=dtype).mode()
+        tm.assert_series_equal(result, expected)

pandas/tests/series/methods/test_value_counts.py

@@ -209,19 +209,25 @@ def test_value_counts_bool_with_nan(self, ser, dropna, exp):
         tm.assert_series_equal(out, exp)
 
     @pytest.mark.parametrize(
-        "input_array,expected",
+        "input_array,expected,dtype",
         [
             (
                 [1 + 1j, 0, 1 + 0j, 1j, 1 + 2j],
                 Series([1, 1, 1, 1, 1], index=[0, 1 + 0j, 1j, 1 + 1j, 1 + 2j]),
+                np.complex128
             ),
             (
                 [1 + 2j, 0, 1j, 1, 1j, 1 + 1j, 1 + 1j],
-                # index is sorted by value counts in descending order by default
                 Series([2, 1, 1, 2, 1], index=pd.Index([1j, 0, 1, 1 + 1j, 1 + 2j], dtype=np.complex128)),
+                np.complex128
+            ),
+            (
+                [1 + 2j, 0, 1j, 1, 1j, 1 + 1j, 1 + 1j],
+                Series([2, 1, 1, 2, 1], index=pd.Index([1j, 0, 1, 1 + 1j, 1 + 2j], dtype=np.complex64)),
+                np.complex64
             ),
         ],
     )
-    def test_value_counts_complex_numbers(self, input_array, expected):
-        result = pd.Series(input_array, dtype=np.complex128).value_counts()
+    def test_value_counts_complex_numbers(self, input_array, expected, dtype):
+        result = pd.Series(input_array, dtype=dtype).value_counts()
         tm.assert_series_equal(result, expected)