Undo some strictness checking

Author: mroeschke

pandas/tests/extension/base/dim2.py

@@ -43,9 +43,7 @@ def test_frame_from_2d_array(self, data):
         arr2d = data.repeat(2).reshape(-1, 2)
 
         df = pd.DataFrame(arr2d)
-        expected = pd.DataFrame(
-            {0: arr2d[:, 0], 1: arr2d[:, 1]}, columns=pd.RangeIndex(2)
-        )
+        expected = pd.DataFrame({0: arr2d[:, 0], 1: arr2d[:, 1]})
         tm.assert_frame_equal(df, expected)
 
     def test_swapaxes(self, data):

pandas/tests/extension/base/reshaping.py

@@ -33,18 +33,18 @@ def test_concat(self, data, in_frame):
 
     @pytest.mark.parametrize("in_frame", [True, False])
     def test_concat_all_na_block(self, data_missing, in_frame):
-        valid_block = pd.Series(data_missing.take([1, 1]), index=[0, 1])
-        na_block = pd.Series(data_missing.take([0, 0]), index=[2, 3])
+        valid_block = pd.Series(data_missing.take([1, 1]), index=range(2))
+        na_block = pd.Series(data_missing.take([0, 0]), index=range(2, 4))
         if in_frame:
             valid_block = pd.DataFrame({"a": valid_block})
             na_block = pd.DataFrame({"a": na_block})
         result = pd.concat([valid_block, na_block])
         if in_frame:
             expected = pd.DataFrame({"a": data_missing.take([1, 1, 0, 0])})
-            tm.assert_frame_equal(result, expected, check_index_type=False)
+            tm.assert_frame_equal(result, expected)
         else:
             expected = pd.Series(data_missing.take([1, 1, 0, 0]))
-            tm.assert_series_equal(result, expected, check_index_type=False)
+            tm.assert_series_equal(result, expected)
 
     def test_concat_mixed_dtypes(self, data):
         # https://github.com/pandas-dev/pandas/issues/20762

pandas/tests/extension/test_arrow.py

@@ -2322,8 +2322,7 @@ def test_str_split():
         {
             0: ArrowExtensionArray(pa.array(["a", "a", None])),
             1: ArrowExtensionArray(pa.array(["cbcb", "cbcb", None])),
-        },
-        columns=range(2),
+        }
     )
     tm.assert_frame_equal(result, expected)
 
@@ -2332,8 +2331,7 @@ def test_str_split():
         {
             0: ArrowExtensionArray(pa.array(["a", "a2cbcb", None])),
             1: ArrowExtensionArray(pa.array(["cbcb", None, None])),
-        },
-        columns=range(2),
+        }
     )
     tm.assert_frame_equal(result, expected)
 
@@ -2358,8 +2356,7 @@ def test_str_rsplit():
         {
             0: ArrowExtensionArray(pa.array(["a1cb", "a2cb", None])),
             1: ArrowExtensionArray(pa.array(["b", "b", None])),
-        },
-        columns=range(2),
+        }
     )
     tm.assert_frame_equal(result, expected)
 
@@ -2368,8 +2365,7 @@ def test_str_rsplit():
         {
             0: ArrowExtensionArray(pa.array(["a", "a2cbcb", None])),
             1: ArrowExtensionArray(pa.array(["cbcb", None, None])),
-        },
-        columns=range(2),
+        }
     )
     tm.assert_frame_equal(result, expected)
 

pandas/tests/frame/test_constructors.py

@@ -89,7 +89,7 @@ def test_constructor_from_2d_datetimearray(self):
 
         df = DataFrame(dta)
         expected = DataFrame({0: dta[:, 0], 1: dta[:, 1]})
-        tm.assert_frame_equal(df, expected, check_column_type=False)
+        tm.assert_frame_equal(df, expected)
         # GH#44724 big performance hit if we de-consolidate
         assert len(df._mgr.blocks) == 1
 
@@ -950,15 +950,15 @@ def test_nested_dict_frame_constructor(self):
                 data.setdefault(col, {})[row] = df._get_value(row, col)
 
         result = DataFrame(data, columns=rng)
-        tm.assert_frame_equal(result, df, check_index_type=False)
+        tm.assert_frame_equal(result, df)
 
         data = {}
         for col in df.columns:
             for row in df.index:
                 data.setdefault(row, {})[col] = df._get_value(row, col)
 
         result = DataFrame(data, index=rng).T
-        tm.assert_frame_equal(result, df, check_index_type=False)
+        tm.assert_frame_equal(result, df)
 
     def _check_basic_constructor(self, empty):
         # mat: 2d matrix with shape (3, 2) to input. empty - makes sized
@@ -1266,10 +1266,10 @@ def test_constructor_list_of_lists(self, using_infer_string):
 
         # GH 4851
         # list of 0-dim ndarrays
-        expected = DataFrame({0: np.arange(10)})
+        expected = DataFrame(np.arange(10))
         data = [np.array(x) for x in range(10)]
         result = DataFrame(data)
-        tm.assert_frame_equal(result, expected, check_column_type=False)
+        tm.assert_frame_equal(result, expected)
 
     def test_nested_pandasarray_matches_nested_ndarray(self):
         # GH#43986
@@ -1403,9 +1403,7 @@ def test_constructor_generator(self):
         gen = ([i, "a"] for i in range(10))
         result = DataFrame(gen)
         expected = DataFrame({0: range(10), 1: "a"})
-        tm.assert_frame_equal(
-            result, expected, check_dtype=False, check_column_type=False
-        )
+        tm.assert_frame_equal(result, expected, check_dtype=False)
 
     def test_constructor_list_of_dicts(self):
         result = DataFrame([{}])
@@ -1651,8 +1649,8 @@ def test_constructor_Series_named(self):
 
         # this is a bit non-intuitive here; the series collapse down to arrays
         df = DataFrame([arr, s1]).T
-        expected = DataFrame({1: s1, 0: arr}, columns=[0, 1])
-        tm.assert_frame_equal(df, expected, check_column_type=False)
+        expected = DataFrame({1: s1, 0: arr}, columns=range(2))
+        tm.assert_frame_equal(df, expected)
 
     def test_constructor_Series_named_and_columns(self):
         # GH 9232 validation
@@ -2187,7 +2185,7 @@ def test_constructor_ndarray_categorical_dtype(self):
         result = DataFrame(arr, dtype=cat.dtype)
 
         expected = DataFrame({0: cat, 1: cat, 2: cat, 3: cat})
-        tm.assert_frame_equal(result, expected, check_column_type=False)
+        tm.assert_frame_equal(result, expected)
 
     def test_constructor_categorical(self):
         # GH8626
@@ -2583,7 +2581,7 @@ def test_from_2d_object_array_of_periods_or_intervals(self):
         data3 = np.r_[data, data2, data, data2].T
         df3 = DataFrame(data3)
         expected = DataFrame({0: pi, 1: ii, 2: pi, 3: ii})
-        tm.assert_frame_equal(df3, expected, check_column_type=False)
+        tm.assert_frame_equal(df3, expected)
 
     @pytest.mark.parametrize(
         "col_a, col_b",
@@ -2676,7 +2674,7 @@ def test_frame_string_inference_array_string_dtype(self):
         expected = DataFrame({0: ["a", "b"], 1: ["c", "d"]}, dtype=dtype)
         with pd.option_context("future.infer_string", True):
             df = DataFrame(np.array([["a", "c"], ["b", "d"]]))
-        tm.assert_frame_equal(df, expected, check_column_type=False)
+        tm.assert_frame_equal(df, expected)
 
         expected = DataFrame(
             {"a": ["a", "b"], "b": ["c", "d"]},

pandas/tests/window/test_dtypes.py

@@ -170,4 +170,4 @@ def test_dataframe_dtypes(method, expected_data, dtypes, min_periods, step):
     else:
         result = getattr(rolled, method)()
         expected = DataFrame(expected_data, dtype="float64")[::step]
-        tm.assert_frame_equal(result, expected, check_column_type=False)
+        tm.assert_frame_equal(result, expected)