All labels to appear in test dataset

Question

I am using model_selection.train_test_split to split a dataset into a training set and test set. I would like to have all labels appear in the test set, although th