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