code: def test_display_dataframe(app: Page):
      """Test that st.write displays pyspark.sql.DataFrame and pd.Dataframe via st.dataframe."""
  
      dataframe_element = app.get_by_test_id("stDataFrame")
      expect(dataframe_element).to_have_count(4)
  
file_path: streamlit/e2e_playwright/st_write_test.py
imports: from playwright.sync_api import Page, expect
label_desc: Use image node_id 2 for calling a global function with example usage: test_display_dataframe(app) without return types