def test_download(name, task, data_dir, extract_data, verbose, is_test): config_cache(reset_cache=True, is_test=True) download(name=name, data_dir=data_dir, extract_data=extract_data, verbose=verbose, is_test=is_test)
def test_load(name, task, data_dir, extract_data, verbose, is_test): config_cache(reset_cache=True, is_test=True) db = load(name=name, task=task, data_dir=data_dir, verbose=verbose, is_test=is_test) assert(db.db_name == name)
def test_add(): config_cache(reset_cache=True, is_test=True) add(name='new_db', task='new_task', data_dir='new/path/db', file_path='newdb.h5', keywords=['new_category'], is_test=True) add('new_db', 'new_task', 'new/path/db', 'newdb.h5', ['new_category'], True)
def test_remove(): add('new_db', 'new_task', 'new/path/db', 'newdb.h5', ['new_category'], True) try: remove(name='new_db', task='new_task', delete_data=True, is_test=True) except Exception: pass config_cache(reset_cache=True, is_test=True)
def test_config_cache(): config_cache(reset_cache=True, is_test=True)