Write Python code to split a dataset into training and test sets using scikit-learn.