add unit tests for sampling, add_rows and remove_rows in transition