def setUpClass(cls):
"""Create a new metadata file if it does not exist.
This is executed before any test is started.
"""
config = open_config(CONF_FILE)
if not os.path.isfile(META_FILE):
sys.stdout.write("Initializing database...\n")
db.make_meta_db(META_FILE)
with db.session_scope(META_FILE) as (session, metadata):
mkmeta = db.MakeMeta(config, IMAGE_DIR)
mkmeta.make(session, metadata)
def setUp(self):
"""Prepare the testing environment."""
self.config = open_config(CONF_FILE)
self.train_file = os.path.join(TEMP_DIR, 'train_data.tsv')
self.ann_file = os.path.join(TEMP_DIR, 'Cypripedium_section.ann')
self.test_result = os.path.join(TEMP_DIR, 'test_result.tsv')
self.train_dir = os.path.join(TEMP_DIR, 'train_data')
self.ann_dir = os.path.join(TEMP_DIR, 'ann_dir')
self.test_result_batch = os.path.join(TEMP_DIR, 'test_result_batch.tsv')
for path in (self.train_dir, self.ann_dir):
if not os.path.isdir(path):
os.mkdir(path)
