def reboot_database():
"""Delete and recreate the database between test classes"""
settings = RaseSettings()
if os.path.isdir(settings.getSampleDirectory()):
shutil.rmtree(settings.getSampleDirectory())
if os.path.isfile(settings.getDatabaseFilepath()):
os.remove(settings.getDatabaseFilepath())
initializeDatabase(settings.getDatabaseFilepath())
def complete_tests():
"""Make sure no sample spectra are left after the final test is run"""
settings = RaseSettings()
original_data_dir = settings.getDataDirectory()
settings.setDataDirectory(os.path.join(os.getcwd(), '__temp_test_rase'))
yield None # anything before this line will be run prior to the tests
print('CLEANING UP') # run after the last test
settings = RaseSettings()
if os.path.isdir(settings.getSampleDirectory()):
shutil.rmtree(settings.getSampleDirectory())
if os.path.isfile(settings.getDatabaseFilepath()):
os.remove(settings.getDatabaseFilepath())
settings.setDataDirectory(original_data_dir)
print('CLEANUP COMPLETE')
