def test_importing_pickles(is_file, glob, mock_open, mock_pickle): from pathlib import Path from menpo.base import LazyList mock_pickle.return_value = {'test': 1} is_file.return_value = True glob.return_value = [Path('mocked1.pkl'), Path('mocked2.pkl')] objs = mio.import_pickles('*') assert isinstance(objs, LazyList) assert len(objs) == 2 assert objs[0]['test'] == 1 assert objs[1]['test'] == 1
def test_importing_pickles_as_generator(is_file, glob, mock_open, mock_pickle): from pathlib import Path import types mock_pickle.return_value = {'test': 1} is_file.return_value = True glob.return_value = [Path('mocked1.pkl'), Path('mocked2.pkl')] objs = mio.import_pickles('*', as_generator=True) assert isinstance(objs, types.GeneratorType) objs = list(objs) assert len(objs) == 2 assert objs[0]['test'] == 1 assert objs[1]['test'] == 1