[MINOR] Change the assertion in test random splitter to a pandas assertion