@@ -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