STY: add strict=True to builtin zip() calls in pandas/tests/indexes/multi (#62634)

Author: divya1974

pandas/tests/indexes/multi/test_analytics.py

@@ -185,7 +185,7 @@ def test_map(idx):
 @pytest.mark.parametrize(
  "mapper",
  [
- lambda values, idx: {i: e for e, i in zip(values, idx)},
+ lambda values, idx: {i: e for e, i in zip(values, idx, strict=True)},
  lambda values, idx: pd.Series(values, idx),
  ],
 )

pandas/tests/indexes/multi/test_constructors.py

@@ -155,7 +155,7 @@ def test_copy_in_constructor():
 def test_from_arrays(idx):
  arrays = [
  np.asarray(lev).take(level_codes)
- for lev, level_codes in zip(idx.levels, idx.codes)
+ for lev, level_codes in zip(idx.levels, idx.codes, strict=True)
  ]
 
  # list of arrays as input
@@ -172,7 +172,7 @@ def test_from_arrays_iterator(idx):
  # GH 18434
  arrays = [
  np.asarray(lev).take(level_codes)
- for lev, level_codes in zip(idx.levels, idx.codes)
+ for lev, level_codes in zip(idx.levels, idx.codes, strict=True)
  ]
 
  # iterator as input
@@ -188,7 +188,7 @@ def test_from_arrays_iterator(idx):
 def test_from_arrays_tuples(idx):
  arrays = tuple(
  tuple(np.asarray(lev).take(level_codes))
- for lev, level_codes in zip(idx.levels, idx.codes)
+ for lev, level_codes in zip(idx.levels, idx.codes, strict=True)
  )
 
  # tuple of tuples as input
@@ -368,7 +368,7 @@ def test_from_tuples_iterator():
  levels=[[1, 3], [2, 4]], codes=[[0, 1], [0, 1]], names=["a", "b"]
  )
 
- result = MultiIndex.from_tuples(zip([1, 3], [2, 4]), names=["a", "b"])
+ result = MultiIndex.from_tuples(zip([1, 3], [2, 4], strict=True), names=["a", "b"])
  tm.assert_index_equal(result, expected)
 
  # input non-iterables

pandas/tests/indexes/multi/test_equivalence.py

@@ -223,7 +223,7 @@ def test_equals_missing_values_differently_sorted():
 
 
 def test_is_():
- mi = MultiIndex.from_tuples(zip(range(10), range(10)))
+ mi = MultiIndex.from_tuples(zip(range(10), range(10), strict=True))
  assert mi.is_(mi)
  assert mi.is_(mi.view())
  assert mi.is_(mi.view().view().view().view())

pandas/tests/indexes/multi/test_get_set.py

@@ -15,7 +15,7 @@ def assert_matching(actual, expected, check_dtype=False):
  # avoid specifying internal representation
  # as much as possible
  assert len(actual) == len(expected)
- for act, exp in zip(actual, expected):
+ for act, exp in zip(actual, expected, strict=True):
  act = np.asarray(act)
  exp = np.asarray(exp)
  tm.assert_numpy_array_equal(act, exp, check_dtype=check_dtype)

pyproject.toml

@@ -520,10 +520,6 @@ exclude = [
 "pandas/tests/indexes/interval/test_constructors.py" = ["B905"]
 "pandas/tests/indexes/interval/test_formats.py" = ["B905"]
 "pandas/tests/indexes/interval/test_interval.py" = ["B905"]
-"pandas/tests/indexes/multi/test_analytics.py" = ["B905"]
-"pandas/tests/indexes/multi/test_constructors.py" = ["B905"]
-"pandas/tests/indexes/multi/test_equivalence.py" = ["B905"]
-"pandas/tests/indexes/multi/test_get_set.py" = ["B905"]
 "pandas/tests/indexes/period/methods/test_asfreq.py" = ["B905"]
 "pandas/tests/indexes/period/test_constructors.py" = ["B905"]
 "pandas/tests/indexes/period/test_formats.py" = ["B905"]