TST: Fix tests to use assert_frame_equal