TST: use DataFrame in the test