Add missing deepcopy import to test_datasets.py