@@ -52,7 +52,12 @@ def test_spectra_viewer_add_from_input(spectra_viewer: Page, assert_snapshot: Ca
5252 # a NEW search input should appear after selecting a protein
5353 tab_wrapper = spectra_viewer .locator (".tab-wrapper" )
5454 expect (tab_wrapper .get_by_text ("Type to search..." )).to_be_visible ()
55- assert_snapshot (spectra_viewer )
55+ if not hasattr (assert_snapshot , "NOOP" ):
56+ # Wait for chart to fully render
57+ spectra_viewer .locator (".highcharts-series" ).first .wait_for (state = "attached" )
58+ spectra_viewer .wait_for_load_state ("networkidle" )
59+ # Mask legend as it has minor rendering variations
60+ assert_snapshot (spectra_viewer , mask_elements = [".highcharts-legend" ])
5661
5762
5863@pytest .mark .parametrize ("method" , ["spacebar" , "click" ])
@@ -90,7 +95,12 @@ def test_spectra_viewer_add_from_spacebar(spectra_viewer: Page, assert_snapshot:
9095 # ... and a selector for EGFP should now be visible
9196 expect (tab_wrapper .get_by_text (re .compile (r"^EGFP" ))).to_be_visible ()
9297
93- assert_snapshot (spectra_viewer )
98+ if not hasattr (assert_snapshot , "NOOP" ):
99+ # Wait for chart to fully render
100+ spectra_viewer .locator (".highcharts-series" ).first .wait_for (state = "attached" )
101+ spectra_viewer .wait_for_load_state ("networkidle" )
102+ # Mask legend as it has minor rendering variations
103+ assert_snapshot (spectra_viewer , mask_elements = [".highcharts-legend" ])
94104
95105
96106def test_spectra_url_sharing_basic (live_server : LiveServer , page : Page ) -> None :
0 commit comments