def setUp(self): test_data_path = path.join( __pkg_root__, 'datasets', 'tests', 'small_dataset' ) preproc = Mock(Preprocessor) preproc.full_set = Preprocessor._get_headers(test_data_path) self.preproc = preproc