improvements

Author: topper-123

pandas/_testing/asserters.py

@@ -317,19 +317,13 @@ def _check_types(left, right, obj="Index") -> None:
 
         # Skip exact dtype checking when `check_categorical` is False
         if is_categorical_dtype(left.dtype) and is_categorical_dtype(right.dtype):
-            if not check_categorical:
+            if check_categorical:
                 assert_attr_equal("dtype", left, right, obj=obj)
                 assert_index_equal(left.categories, right.categories, exact=exact)
-            else:
-                assert_attr_equal("inferred_type", left, right, obj=obj)
-            return
-
-        # allow string-like to have different inferred_types
-        if left.inferred_type in ("string"):
-            assert right.inferred_type in ("string")
-        else:
             assert_attr_equal("inferred_type", left, right, obj=obj)
+            return
 
+        assert_attr_equal("inferred_type", left, right, obj=obj)
         assert_attr_equal("dtype", left, right, obj=obj)
 
     def _get_ilevel_values(index, level):
@@ -457,7 +451,7 @@ def repr_class(x):
         return
 
     if exact == "equiv":
-        # accept equivalence of all NumericIndex (sub-)classes
+        # accept equivalence of NumericIndex (sub-)classes
         if isinstance(left, NumericIndex) and isinstance(right, NumericIndex):
             return
 

pandas/tests/util/test_assert_index_equal.py

@@ -58,15 +58,30 @@ def test_index_equal_length_mismatch(check_exact):
         tm.assert_index_equal(idx1, idx2, check_exact=check_exact)
 
 
-def test_index_equal_class_mismatch(check_exact):
-    msg = """Index are different
+@pytest.mark.parametrize("exact", [False, "equiv"])
+def test_index_equal_class(exact):
+    idx1 = Index([0, 1, 2])
+    idx2 = RangeIndex(3)
+
+    tm.assert_index_equal(idx1, idx2, exact=exact)
+
+
+@pytest.mark.parametrize(
+    "idx_values, msg_str",
+    [
+        [[1, 2, 3.0], "Float64Index\\(\\[1\\.0, 2\\.0, 3\\.0\\], dtype='float64'\\)"],
+        [range(3), "RangeIndex\\(start=0, stop=3, step=1\\)"],
+    ],
+)
+def test_index_equal_class_mismatch(check_exact, idx_values, msg_str):
+    msg = f"""Index are different
 
 Index classes are different
 \\[left\\]:  Int64Index\\(\\[1, 2, 3\\], dtype='int64'\\)
-\\[right\\]: Float64Index\\(\\[1\\.0, 2\\.0, 3\\.0\\], dtype='float64'\\)"""
+\\[right\\]: {msg_str}"""
 
     idx1 = Index([1, 2, 3])
-    idx2 = Index([1, 2, 3.0])
+    idx2 = Index(idx_values)
 
     with pytest.raises(AssertionError, match=msg):
         tm.assert_index_equal(idx1, idx2, exact=True, check_exact=check_exact)