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