def test_gen_orig_params(self, mock_gen):
mock_gen.return_value = [
None, ['path/to/sra', 'path/to/sra.download.COMPLETE'],
'mock_sample'
]
mock_samples = ['mock_sample'] * 2
self.assertEqual(download.gen_orig_params(mock_samples), [
None, ['path/to/sra', 'path/to/sra.download.COMPLETE'],
'mock_sample', None,
['path/to/sra', 'path/to/sra.download.COMPLETE'], 'mock_sample'
])
def originate_params():
"""
Generate a list of sras to download for each sample
This function gets called twice, once before entering the queue, once after
"""
num_samples = len(samples)
logger.info('preparing originate_params for {0} samples'.format(num_samples))
orig_params_sets = gen_orig_params(samples)
logger.info(
'{0} sets of orig_params generated'.format(len(orig_params_sets)))
for _ in orig_params_sets:
yield _
def originate_params():
"""
Generate a list of sras to download for each sample
This function gets called twice, once before entering the queue, once after
"""
num_samples = len(samples)
logger.info(
'preparing originate_params for {0} samples'.format(num_samples))
orig_params_sets = gen_orig_params(samples)
logger.info('{0} sets of orig_params generated'.format(
len(orig_params_sets)))
for _ in orig_params_sets:
yield _
def test_gen_orig_params(self, mock_gen):
mock_gen.return_value = [None, ['path/to/sra', 'path/to/sra.download.COMPLETE'], 'mock_sample']
mock_samples = ['mock_sample'] * 2
self.assertEqual(download.gen_orig_params(mock_samples),
[None, ['path/to/sra', 'path/to/sra.download.COMPLETE'], 'mock_sample',
None, ['path/to/sra', 'path/to/sra.download.COMPLETE'], 'mock_sample'])