if train_df.shape[0] + test_df.shape[0] == train_info_df.shape[0]: