Skip to content

Commit c8a460a

Browse files
fix name issue
1 parent 77d20e3 commit c8a460a

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

bigframes/core/blocks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -445,7 +445,7 @@ def reset_index(self, level: LevelsType = None, drop: bool = True) -> Block:
445445
if "index" not in self.column_labels and self.index.nlevels <= 1:
446446
label = "index"
447447
else:
448-
label = f"level_{self.index.names.index(label)}"
448+
label = f"level_{self.index_columns.index(level_id)}"
449449

450450
if label in self.column_labels:
451451
raise ValueError(f"cannot insert {label}, already exists")

tests/unit/test_dataframe_polars.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1657,13 +1657,11 @@ def test_reset_index_with_unnamed_index(
16571657
pandas.testing.assert_frame_equal(bf_result, pd_result)
16581658

16591659

1660-
def test_reset_index_with_unnamed_multiindex(
1661-
scalars_df_index,
1662-
scalars_pandas_df_index,
1663-
):
1660+
def test_reset_index_with_unnamed_multiindex(session):
16641661
bf_df = dataframe.DataFrame(
16651662
([1, 2, 3], [2, 5, 7]),
16661663
index=pd.MultiIndex.from_tuples([("a", "aa"), ("a", "aa")]),
1664+
session=session,
16671665
)
16681666
pd_df = pd.DataFrame(
16691667
([1, 2, 3], [2, 5, 7]),

0 commit comments

Comments
 (0)