def test_iter_dir(self, args, blast_db, query_file, iteration):
assume(re.search(r'\w', query_file))
assembler = BaseAssembler(args, 'db_conn')
assembler.set_state(blast_db, query_file, iteration)
base_blast_db = basename(blast_db)
base_query_file = basename(query_file) if iteration == 1 else ''
dir_name = '{}_{}_{:02d}'.format(base_blast_db, base_query_file,
iteration)
expect = join(args['temp_dir'], dir_name)
assert expect == assembler.iter_dir()
def test_work_path(self):
iter_dir = 'whatever'
assembler = BaseAssembler('args', 'db_conn')
assembler.iter_dir = MagicMock(return_value=iter_dir)
assert iter_dir == assembler.work_path()