Skip to content

Commit 7fdb68e

Browse files
committed
fix: Add clean_formatter_state fixture to session display config tests
1 parent f4c5003 commit 7fdb68e

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

python/tests/test_dataframe.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -131,11 +131,11 @@ def test_session_with_display_config(data):
131131
assert html_repr != html_repr2
132132

133133
# Check that the second representation has the short cell data based on the configured length
134-
assert f'<span class="expandable" id="' in html_repr2
134+
assert f"<span class='expandable' id=" in html_repr2
135135
assert f'>{("x" * 10)}</span>' in html_repr2
136136

137137

138-
def test_display_config_in_init(data):
138+
def test_display_config_in_init(data, clean_formatter_state):
139139
# Test providing display config directly in SessionContext constructor
140140
display_config = DataframeDisplayConfig(
141141
max_table_bytes=1024,
@@ -1789,7 +1789,7 @@ def test_display_config_affects_repr(data):
17891789
assert "Data truncated" not in repr_str2
17901790

17911791

1792-
def test_display_config_affects_html_repr(data):
1792+
def test_display_config_affects_html_repr(data, clean_formatter_state):
17931793
# Create a context with custom display config to show only a small cell length
17941794
ctx = SessionContext().with_display_config(max_cell_length=5)
17951795

@@ -1801,7 +1801,7 @@ def test_display_config_affects_html_repr(data):
18011801

18021802
# The cell should be truncated to 5 characters and have expansion button
18031803
assert ">xxxxx" in html_str # 5 character limit
1804-
expandable_class = 'class="expandable-container"'
1804+
expandable_class = '.expandable-container'
18051805
assert expandable_class in html_str
18061806

18071807
# Create a context with larger cell length limit

0 commit comments

Comments
 (0)