update tests

Author: topper-123

pandas/_testing/asserters.py

@@ -16,8 +16,6 @@
     is_bool,
     is_categorical_dtype,
     is_extension_array_dtype,
-    is_float_dtype,
-    is_integer_dtype,
     is_interval_dtype,
     is_number,
     is_numeric_dtype,

pandas/tests/indexes/common.py

@@ -13,6 +13,7 @@
     is_datetime64tz_dtype,
     is_float_dtype,
     is_integer_dtype,
+    is_unsigned_integer_dtype,
 )
 from pandas.core.dtypes.dtypes import CategoricalDtype
 
@@ -29,7 +30,6 @@
     RangeIndex,
     Series,
     TimedeltaIndex,
-    UInt64Index,
     isna,
 )
 import pandas._testing as tm
@@ -655,7 +655,7 @@ def test_map_dictlike(self, mapper, simple_index):
         identity = mapper(idx.values, idx)
 
         # we don't infer to UInt64 for a dict
-        if isinstance(idx, UInt64Index) and isinstance(identity, dict):
+        if is_unsigned_integer_dtype(idx.dtype) and isinstance(identity, dict):
             expected = idx.astype("int64")
         else:
             expected = idx

pandas/tests/indexes/numeric/test_numeric.py

@@ -431,17 +431,6 @@ def index(self, request, dtype):
     def test_constructor(self, dtype):
         index_cls = self._index_cls
 
-        # pass list, coerce fine
-        index = index_cls([-5, 0, 1, 2], dtype=dtype)
-        expected = Index([-5, 0, 1, 2], dtype=dtype)
-        exact = True if index_cls is Int64Index else "equiv"
-        tm.assert_index_equal(index, expected, exact=exact)
-
-        # from iterable
-        index = index_cls(iter([-5, 0, 1, 2]), dtype=dtype)
-        expected = index_cls([-5, 0, 1, 2], dtype=dtype)
-        tm.assert_index_equal(index, expected, exact=True)
-
         # scalar raise Exception
         msg = (
             rf"{index_cls.__name__}\(\.\.\.\) must be called with a collection of some "
@@ -451,6 +440,8 @@ def test_constructor(self, dtype):
             index_cls(5)
 
         # copy
+        # pass list, coerce fine
+        index = index_cls([-5, 0, 1, 2], dtype=dtype)
         arr = index.values
         new_index = index_cls(arr, copy=True)
         tm.assert_index_equal(new_index, index, exact=True)
@@ -460,25 +451,38 @@ def test_constructor(self, dtype):
         arr[0] = val
         assert new_index[0] != val
 
-        # interpret list-like
-        expected = index_cls([5, 0])
-        for cls in [Index, index_cls]:
-            exact = True if cls is Int64Index else "equiv"
-            for idx in [
-                cls([5, 0], dtype=dtype),
-                cls(np.array([5, 0]), dtype=dtype),
-                cls(Series([5, 0]), dtype=dtype),
-            ]:
-                tm.assert_index_equal(idx, expected, exact=exact)
+        if dtype == np.int64:
+            exact = "equiv" if index_cls != Int64Index else True
+
+            # pass list, coerce fine
+            index = index_cls([-5, 0, 1, 2], dtype=dtype)
+            expected = Index([-5, 0, 1, 2], dtype=dtype)
+            tm.assert_index_equal(index, expected, exact=exact)
+
+            # from iterable
+            index = index_cls(iter([-5, 0, 1, 2]), dtype=dtype)
+            expected = index_cls([-5, 0, 1, 2], dtype=dtype)
+            tm.assert_index_equal(index, expected, exact=exact)
+
+            # interpret list-like
+            expected = index_cls([5, 0], dtype=dtype)
+            for cls in [Index, index_cls]:
+                for idx in [
+                    cls([5, 0], dtype=dtype),
+                    cls(np.array([5, 0]), dtype=dtype),
+                    cls(Series([5, 0]), dtype=dtype),
+                ]:
+                    tm.assert_index_equal(idx, expected, exact=exact)
 
     def test_constructor_corner(self, dtype):
         index_cls = self._index_cls
 
         arr = np.array([1, 2, 3, 4], dtype=object)
         index = index_cls(arr, dtype=dtype)
         assert index.values.dtype == index.dtype
-        exact = True if index_cls is Int64Index else "equiv"
-        tm.assert_index_equal(index, Index(arr), exact=exact)
+        if dtype == np.int64:
+            exact = True if index_cls is Int64Index else "equiv"
+            tm.assert_index_equal(index, Index(arr), exact=exact)
 
         # preventing casting
         arr = np.array([1, "2", 3, "4"], dtype=object)

pandas/tests/indexes/test_any_index.py

@@ -5,8 +5,11 @@
 """
 import re
 
+import numpy as np
 import pytest
 
+from pandas.core.dtypes.common import is_float_dtype
+
 import pandas._testing as tm
 
 
@@ -47,7 +50,17 @@ def test_mutability(index):
 
 def test_map_identity_mapping(index):
     # GH#12766
-    tm.assert_index_equal(index, index.map(lambda x: x))
+    result = index.map(lambda x: x)
+    if index._is_numeric_index:
+        if is_float_dtype(index.dtype):
+            expected = index.astype(np.float64)
+        elif index.dtype == np.uint64:
+            expected = index.astype(np.uint64)
+        else:
+            expected = index.astype(np.int64)
+    else:
+        expected = index
+    tm.assert_index_equal(result, expected)
 
 
 def test_wrong_number_names(index):

pandas/tests/indexes/test_base.py

@@ -35,6 +35,7 @@
     period_range,
 )
 import pandas._testing as tm
+from pandas.api.types import is_float_dtype
 from pandas.core.indexes.api import (
     Index,
     MultiIndex,
@@ -714,7 +715,11 @@ def test_map_dictlike(self, index, mapper):
             # to match proper result coercion for uints
             expected = Index([])
         elif index._is_numeric_index:
-            expected = type(index)(np.arange(len(index), 0, -1), dtype=index.dtype)
+            if is_float_dtype(index.dtype):
+                exp_dtype = np.float64
+            else:
+                exp_dtype = np.int64
+            expected = index._constructor(np.arange(len(index), 0, -1), dtype=exp_dtype)
         else:
             expected = Index(np.arange(len(index), 0, -1))