From fed8820e86f031ba68bebfec2bb97dcff3bf606f Mon Sep 17 00:00:00 2001 From: Shawn Liu Date: Sat, 20 Jul 2024 20:18:07 -0400 Subject: [PATCH 1/2] Add fixture docstring for series indexing --- pandas/tests/series/indexing/test_setitem.py | 23 ++++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/pandas/tests/series/indexing/test_setitem.py b/pandas/tests/series/indexing/test_setitem.py index 253339f8a6446..cab16830fd44c 100644 --- a/pandas/tests/series/indexing/test_setitem.py +++ b/pandas/tests/series/indexing/test_setitem.py @@ -1178,25 +1178,48 @@ def test_setitem_example(self): @pytest.fixture def obj(self): + """ + Fixture to create a Series instance using IntervalIndex + [(0, 1], (1, 2], (2, 3]] + """ idx = IntervalIndex.from_breaks(range(4)) return Series(idx) @pytest.fixture def val(self): + """ + Fixture to get an interval (0.5, 1.5] + """ return Interval(0.5, 1.5) @pytest.fixture def key(self): + """ + Fixture to get a key 0 + """ return 0 @pytest.fixture def expected(self, obj, val): + """ + Fixture to get a Series instance with dtype:float64 using the given + input fixture obj and fixture val. + Parameters: + obj(Series): [(0, 1], (1, 2], (2, 3]] + val(Interval): (0.5, 1.5] + + Returns: + (Series): [(0.5, 1.5], (1.0, 2.0], (2.0, 3.0]] + """ data = [val] + list(obj[1:]) idx = IntervalIndex(data, dtype="Interval[float64]") return Series(idx) @pytest.fixture def raises(self): + """ + Fixture to enable raising pytest exceptions + """ return True From 6515aa6ff422a9ce5e3830764eb04f5945fc064b Mon Sep 17 00:00:00 2001 From: Shawn Liu Date: Mon, 22 Jul 2024 20:53:16 -0400 Subject: [PATCH 2/2] Make fixture docstring one line only --- pandas/tests/series/indexing/test_setitem.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/pandas/tests/series/indexing/test_setitem.py b/pandas/tests/series/indexing/test_setitem.py index cab16830fd44c..62f2c93ef691a 100644 --- a/pandas/tests/series/indexing/test_setitem.py +++ b/pandas/tests/series/indexing/test_setitem.py @@ -1179,8 +1179,7 @@ def test_setitem_example(self): @pytest.fixture def obj(self): """ - Fixture to create a Series instance using IntervalIndex - [(0, 1], (1, 2], (2, 3]] + Fixture to create a Series [(0, 1], (1, 2], (2, 3]] """ idx = IntervalIndex.from_breaks(range(4)) return Series(idx) @@ -1202,14 +1201,7 @@ def key(self): @pytest.fixture def expected(self, obj, val): """ - Fixture to get a Series instance with dtype:float64 using the given - input fixture obj and fixture val. - Parameters: - obj(Series): [(0, 1], (1, 2], (2, 3]] - val(Interval): (0.5, 1.5] - - Returns: - (Series): [(0.5, 1.5], (1.0, 2.0], (2.0, 3.0]] + Fixture to get a Series [(0.5, 1.5], (1.0, 2.0], (2.0, 3.0]] """ data = [val] + list(obj[1:]) idx = IntervalIndex(data, dtype="Interval[float64]")