More tests

alimcmaster1 · alimcmaster1 · commit 6b4c10e87716 · 2021-08-26T22:55:38.000+01:00
diff --git a/pandas/tests/series/methods/test_value_counts.py b/pandas/tests/series/methods/test_value_counts.py
@@ -213,11 +213,15 @@ def test_value_counts_bool_with_nan(self, ser, dropna, exp):
         [
             (
                 [1 + 1j, 1 + 1j, 1, 3j, 3j, 3j],
-                Series([3, 2, 1], index=pd.Index([3j, 1 + 1j, 1])),
-            )
+                Series([3, 2, 1], index=pd.Index([3j, 1 + 1j, 1], dtype=np.complex128)),
+            ),
+            (
+                [1 + 1j, 1 + 1j, 1, 3j, 3j, 3j],
+                Series([3, 2, 1], index=pd.Index([3j, 1 + 1j, 1], dtype=np.complex64)),
+            ),
         ],
     )
     def test_value_counts_complex_numbers(self, input_array, expected):
         # Complex Index dtype is cast to object
         result = Series(input_array).value_counts()
-        tm.assert_series_equal(result, expected, check_index_type=False)
+        tm.assert_series_equal(result, expected)

Original file line number	Diff line number	Diff line change
`@@ -213,11 +213,15 @@ def test_value_counts_bool_with_nan(self, ser, dropna, exp):`
`213`	`213`	`[`
`214`	`214`	`(`
`215`	`215`	`[1 + 1j, 1 + 1j, 1, 3j, 3j, 3j],`
`216`		`- Series([3, 2, 1], index=pd.Index([3j, 1 + 1j, 1])),`
`217`		`- )`
	`216`	`+ Series([3, 2, 1], index=pd.Index([3j, 1 + 1j, 1], dtype=np.complex128)),`
	`217`	`+ ),`
	`218`	`+ (`
	`219`	`+ [1 + 1j, 1 + 1j, 1, 3j, 3j, 3j],`
	`220`	`+ Series([3, 2, 1], index=pd.Index([3j, 1 + 1j, 1], dtype=np.complex64)),`
	`221`	`+ ),`
`218`	`222`	`],`
`219`	`223`	`)`
`220`	`224`	`def test_value_counts_complex_numbers(self, input_array, expected):`
`221`	`225`	`# Complex Index dtype is cast to object`
`222`	`226`	`result = Series(input_array).value_counts()`
`223`		`- tm.assert_series_equal(result, expected, check_index_type=False)`
	`227`	`+ tm.assert_series_equal(result, expected)`