Migrate dataframe tests to insta (#15262)

* convert to assert_snapshot

* edit comments

* formatting

* formatting

* ran cargo fmt --all

* add get_batches_with_limit function

Author: jsai28