From 2a781d7b9669228a79725bac57192848f3c1579f Mon Sep 17 00:00:00 2001 From: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com> Date: Tue, 28 Nov 2023 09:43:14 -0800 Subject: [PATCH] Remove makeIntIndex/makeNumericIndex --- pandas/_testing/__init__.py | 30 ------------------- pandas/conftest.py | 16 ++++++---- pandas/tests/indexes/test_base.py | 6 ++-- pandas/tests/series/indexing/test_setitem.py | 4 ++- .../series/methods/test_combine_first.py | 4 +-- 5 files changed, 18 insertions(+), 42 deletions(-) diff --git a/pandas/_testing/__init__.py b/pandas/_testing/__init__.py index 1d96772b011dc..b1918e1b1d7c2 100644 --- a/pandas/_testing/__init__.py +++ b/pandas/_testing/__init__.py @@ -28,10 +28,7 @@ from pandas.core.dtypes.common import ( is_sequence, - is_signed_integer_dtype, is_string_dtype, - is_unsigned_integer_dtype, - pandas_dtype, ) import pandas as pd @@ -351,31 +348,6 @@ def getCols(k) -> str: return string.ascii_uppercase[:k] -def makeNumericIndex(k: int = 10, *, name=None, dtype: Dtype | None) -> Index: - dtype = pandas_dtype(dtype) - assert isinstance(dtype, np.dtype) - - if dtype.kind in "iu": - values = np.arange(k, dtype=dtype) - if is_unsigned_integer_dtype(dtype): - values += 2 ** (dtype.itemsize * 8 - 1) - elif dtype.kind == "f": - values = np.random.default_rng(2).random(k) - np.random.default_rng(2).random(1) - values.sort() - values = values * (10 ** np.random.default_rng(2).integers(0, 9)) - else: - raise NotImplementedError(f"wrong dtype {dtype}") - - return Index(values, dtype=dtype, name=name) - - -def makeIntIndex(k: int = 10, *, name=None, dtype: Dtype = "int64") -> Index: - dtype = pandas_dtype(dtype) - if not is_signed_integer_dtype(dtype): - raise TypeError(f"Wrong dtype {dtype}") - return makeNumericIndex(k, name=name, dtype=dtype) - - def makeDateIndex( k: int = 10, freq: Frequency = "B", name=None, **kwargs ) -> DatetimeIndex: @@ -962,8 +934,6 @@ def shares_memory(left, right) -> bool: "makeCustomIndex", "makeDataFrame", "makeDateIndex", - "makeIntIndex", - "makeNumericIndex", "makeObjectSeries", "makeTimeDataFrame", "makeTimeSeries", diff --git a/pandas/conftest.py b/pandas/conftest.py index 013b500d95c97..1bc067eb32aef 100644 --- a/pandas/conftest.py +++ b/pandas/conftest.py @@ -620,10 +620,10 @@ def _create_mi_with_dt64tz_level(): "period": period_range("2020-01-01", periods=100, freq="D"), "timedelta": timedelta_range(start="1 day", periods=100, freq="D"), "range": RangeIndex(100), - "int8": tm.makeIntIndex(100, dtype="int8"), - "int16": tm.makeIntIndex(100, dtype="int16"), - "int32": tm.makeIntIndex(100, dtype="int32"), - "int64": tm.makeIntIndex(100, dtype="int64"), + "int8": Index(np.arange(100), dtype="int8"), + "int16": Index(np.arange(100), dtype="int16"), + "int32": Index(np.arange(100), dtype="int32"), + "int64": Index(np.arange(100), dtype="int64"), "uint8": Index(np.arange(100), dtype="uint8"), "uint16": Index(np.arange(100), dtype="uint16"), "uint32": Index(np.arange(100), dtype="uint32"), @@ -632,8 +632,12 @@ def _create_mi_with_dt64tz_level(): "float64": Index(np.arange(100), dtype="float64"), "bool-object": Index([True, False] * 5, dtype=object), "bool-dtype": Index(np.random.default_rng(2).standard_normal(10) < 0), - "complex64": tm.makeNumericIndex(100, dtype="float64").astype("complex64"), - "complex128": tm.makeNumericIndex(100, dtype="float64").astype("complex128"), + "complex64": Index( + np.arange(100, dtype="complex64") + 1.0j * np.arange(100, dtype="complex64") + ), + "complex128": Index( + np.arange(100, dtype="complex128") + 1.0j * np.arange(100, dtype="complex128") + ), "categorical": CategoricalIndex(list("abcd") * 25), "interval": IntervalIndex.from_breaks(np.linspace(0, 100, num=101)), "empty": Index([]), diff --git a/pandas/tests/indexes/test_base.py b/pandas/tests/indexes/test_base.py index 662f31cc3560e..3db81c0285bd2 100644 --- a/pandas/tests/indexes/test_base.py +++ b/pandas/tests/indexes/test_base.py @@ -507,8 +507,8 @@ def test_map_with_tuples(self): # Test that returning a single tuple from an Index # returns an Index. - index = tm.makeIntIndex(3) - result = tm.makeIntIndex(3).map(lambda x: (x,)) + index = Index(np.arange(3), dtype=np.int64) + result = index.map(lambda x: (x,)) expected = Index([(i,) for i in index]) tm.assert_index_equal(result, expected) @@ -555,7 +555,7 @@ def test_map_tseries_indices_accsr_return_index(self): def test_map_dictlike_simple(self, mapper): # GH 12756 expected = Index(["foo", "bar", "baz"]) - index = tm.makeIntIndex(3) + index = Index(np.arange(3), dtype=np.int64) result = index.map(mapper(expected.values, index)) tm.assert_index_equal(result, expected) diff --git a/pandas/tests/series/indexing/test_setitem.py b/pandas/tests/series/indexing/test_setitem.py index 0f3577a214186..cac7bd6de9d3b 100644 --- a/pandas/tests/series/indexing/test_setitem.py +++ b/pandas/tests/series/indexing/test_setitem.py @@ -237,7 +237,9 @@ def test_setitem_slice_integers(self): def test_setitem_slicestep(self): # caught this bug when writing tests - series = Series(tm.makeIntIndex(20).astype(float), index=tm.makeIntIndex(20)) + series = Series( + np.arange(20, dtype=np.float64), index=np.arange(20, dtype=np.int64) + ) series[::2] = 0 assert (series[::2] == 0).all() diff --git a/pandas/tests/series/methods/test_combine_first.py b/pandas/tests/series/methods/test_combine_first.py index 1cbb7c7982802..e1ec8afda33a9 100644 --- a/pandas/tests/series/methods/test_combine_first.py +++ b/pandas/tests/series/methods/test_combine_first.py @@ -32,8 +32,8 @@ def test_combine_first_name(self, datetime_series): assert result.name == datetime_series.name def test_combine_first(self): - values = tm.makeIntIndex(20).values.astype(float) - series = Series(values, index=tm.makeIntIndex(20)) + values = np.arange(20, dtype=np.float64) + series = Series(values, index=np.arange(20, dtype=np.int64)) series_copy = series * 2 series_copy[::2] = np.nan