code: def test_then_remove_rows(dataframe):
      df = dataframe.then(remove_rows_3_and_4)
      rows = tuple(df.index)
      assert rows == (0, 1, 2, 5, 6, 7, 8)
  
imports: import pytest
label_desc: Use image node_id 4 for calling a global function with example usage: test_then_remove_rows(dataframe) without return types