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(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
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
