train_test_split without using sklearn.model_selection