Skip to content

Commit 97a98e5

Browse files
author
cloudboat
committed
STYLE: Format code with ruff
1 parent 5f0caf0 commit 97a98e5

File tree

2 files changed

+9
-12
lines changed

2 files changed

+9
-12
lines changed

pandas/core/indexes/multi.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2731,11 +2731,11 @@ def insert_level(self, position: int, value, name=None) -> MultiIndex:
27312731
27322732
Examples
27332733
--------
2734-
>>> idx = pd.MultiIndex.from_tuples([('A', 1), ('B', 2)], names=['x', 'y'])
2735-
>>> idx.insert_level(0, 'grp')
2734+
>>> idx = pd.MultiIndex.from_tuples([("A", 1), ("B", 2)], names=["x", "y"])
2735+
>>> idx.insert_level(0, "grp")
27362736
MultiIndex([('grp', 'A', 1), ('grp', 'B', 2)],
27372737
names=[None, 'x', 'y'])
2738-
>>> idx.insert_level(1, ['L1', 'L2'], name='z')
2738+
>>> idx.insert_level(1, ["L1", "L2"], name="z")
27392739
MultiIndex([('A', 'L1', 1), ('B', 'L2', 2)],
27402740
names=['x', 'z', 'y'])
27412741
"""
@@ -2773,6 +2773,7 @@ def insert_level(self, position: int, value, name=None) -> MultiIndex:
27732773
new_names.insert(position, name)
27742774

27752775
from pandas import MultiIndex
2776+
27762777
return MultiIndex.from_tuples(new_tuples, names=new_names)
27772778

27782779
def _reorder_ilevels(self, order) -> MultiIndex:
@@ -4451,5 +4452,3 @@ def cartesian_product(X: list[np.ndarray]) -> list[np.ndarray]:
44514452
)
44524453
for i, x in enumerate(X)
44534454
]
4454-
4455-

pandas/tests/indexes/multi/test_insert_level.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55

66

77
class TestMultiIndexInsertLevel:
8-
98
def setup_method(self):
109
self.simple_idx = pd.MultiIndex.from_tuples(
1110
[("A", 1), ("B", 2), ("C", 3)], names=["level1", "level2"]
@@ -16,14 +15,14 @@ def test_insert_level_basic(self):
1615
result = self.simple_idx.insert_level(0, "new_value")
1716
expected = pd.MultiIndex.from_tuples(
1817
[("new_value", "A", 1), ("new_value", "B", 2), ("new_value", "C", 3)],
19-
names=[None, "level1", "level2"]
18+
names=[None, "level1", "level2"],
2019
)
2120
tm.assert_index_equal(result, expected)
2221

2322
result = self.simple_idx.insert_level(1, "middle")
2423
expected = pd.MultiIndex.from_tuples(
2524
[("A", "middle", 1), ("B", "middle", 2), ("C", "middle", 3)],
26-
names=["level1", None, "level2"]
25+
names=["level1", None, "level2"],
2726
)
2827
tm.assert_index_equal(result, expected)
2928

@@ -32,7 +31,7 @@ def test_insert_level_with_different_values(self):
3231
result = self.simple_idx.insert_level(1, new_values)
3332
expected = pd.MultiIndex.from_tuples(
3433
[("A", "X", 1), ("B", "Y", 2), ("C", "Z", 3)],
35-
names=["level1", None, "level2"]
34+
names=["level1", None, "level2"],
3635
)
3736
tm.assert_index_equal(result, expected)
3837

@@ -78,8 +77,7 @@ def test_insert_level_preserves_original(self):
7877

7978
def test_debug_names():
8079
idx = pd.MultiIndex.from_tuples(
81-
[("A", 1), ("B", 2), ("C", 3)],
82-
names=["level1", "level2"]
80+
[("A", 1), ("B", 2), ("C", 3)], names=["level1", "level2"]
8381
)
8482
print("Original names:", idx.names)
8583

@@ -88,6 +86,6 @@ def test_debug_names():
8886

8987
expected = pd.MultiIndex.from_tuples(
9088
[("new_value", "A", 1), ("new_value", "B", 2), ("new_value", "C", 3)],
91-
names=[None, "level1", "level2"]
89+
names=[None, "level1", "level2"],
9290
)
9391
print("Expected names:", expected.names)

0 commit comments

Comments
 (0)