Adds restore_test_data method.