def test_remove_thumbs(self):
"""Verify remove_thumbs method deletes all files in the thumbnail
folder"""
shutil.copy(__file__, pathfinder.thumbnails_path())
self.assertTrue(len(os.listdir(pathfinder.thumbnails_path())) > 0)
self.model.remove_thumbs()
self.assertListEqual(os.listdir(pathfinder.thumbnails_path()), [])
def test_thumbnail_name(self):
"""Verify thumbnail_name function returns the proper thumbnail filename"""
m = hashlib.md5(self.sample_data_file)
expected_thumb_name = os.path.join(pathfinder.thumbnails_path(),
m.hexdigest() + '.png')
self.assertEqual(expected_thumb_name,
model.thumbnail_name(self.sample_data_file))
def check_user_path(self):
"""Verify user data folders were created"""
data_folders = [pathfinder.user_path(), pathfinder.data_path(),
pathfinder.thumbnails_path(), pathfinder.gates_path(),
pathfinder.plugins_path(), pathfinder.podmodels_path(),
pathfinder.colormaps_path(), pathfinder.batchoutput_path()]
self.model.check_user_path()
for folder in data_folders:
self.assertTrue(os.path.exists(folder))
def check_user_path(self):
"""Verify user data folders were created"""
data_folders = [pathfinder.user_path(), pathfinder.data_path(),
pathfinder.thumbnails_path(), pathfinder.gates_path(),
pathfinder.plugins_path(), pathfinder.podmodels_path(),
pathfinder.adamodels_path(), pathfinder.colormaps_path(),
pathfinder.batchoutput_path()]
self.model.check_user_path()
for folder in data_folders:
self.assertTrue(os.path.exists(folder))
def check_user_path(cls):
"""Verify that user data folders exist. Creates
any missing folders."""
user_folder = pathfinder.user_path()
data_folder = pathfinder.data_path()
thumbnail_folder = pathfinder.thumbnails_path()
plugins_folder = pathfinder.plugins_path()
podmodels_folder = pathfinder.podmodels_path()
adamodels_folder = pathfinder.adamodels_path()
gates_folder = pathfinder.gates_path()
colormaps_folder = pathfinder.colormaps_path()
batch_folder = pathfinder.batchoutput_path()
for fldr in (user_folder, data_folder, thumbnail_folder, plugins_folder, podmodels_folder, gates_folder,
adamodels_folder, colormaps_folder, batch_folder):
if not os.path.exists(fldr):
os.makedirs(fldr)
def check_user_path(cls):
"""Verify that user data folders exist. Creates
any missing folders."""
user_folder = pathfinder.user_path()
data_folder = pathfinder.data_path()
thumbnail_folder = pathfinder.thumbnails_path()
plugins_folder = pathfinder.plugins_path()
podmodels_folder = pathfinder.podmodels_path()
gates_folder = pathfinder.gates_path()
colormaps_folder = pathfinder.colormaps_path()
batch_folder = pathfinder.batchoutput_path()
for fldr in (user_folder, data_folder, thumbnail_folder,
plugins_folder, podmodels_folder, gates_folder,
colormaps_folder, batch_folder):
if not os.path.exists(fldr):
os.makedirs(fldr)
def deleted_user_path():
"""Utility function to delete empty folders in the user data folders,
used to verify that MainModel will recreate missing folders as required.
Returns a list of folders successfully deleted or None if no folders
were deleted."""
data_folders = [pathfinder.user_path(), pathfinder.data_path(), pathfinder.thumbnails_path(),
pathfinder.plugins_path(), pathfinder.colormaps_path()]
deleted_folders = []
for folder in data_folders:
exists_and_empty = os.path.exists(folder) and os.listdir(folder) == []
if exists_and_empty:
try:
os.rmdir(folder)
deleted_folders.append(folder)
except WindowsError: # folder in use (Explorer, cmd, etc.)
pass
if deleted_folders:
return deleted_folders
return None
def deleted_user_path():
"""Utility function to delete empty folders in the user data folders,
used to verify that MainModel will recreate missing folders as required.
Returns a list of folders successfully deleted or None if no folders
were deleted."""
data_folders = [pathfinder.user_path(), pathfinder.data_path(), pathfinder.thumbnails_path(),
pathfinder.plugins_path(), pathfinder.podmodels_path(), pathfinder.adamodels_path(),
pathfinder.colormaps_path()]
deleted_folders = []
for folder in data_folders:
exists_and_empty = os.path.exists(folder) and os.listdir(folder) == []
if exists_and_empty:
try:
os.rmdir(folder)
deleted_folders.append(folder)
except WindowsError: # folder in use (Explorer, cmd, etc.)
pass
if deleted_folders:
return deleted_folders
return None
def remove_thumbs(self):
"""Removes all the thumnbail plots"""
thumbnail_path = pathfinder.thumbnails_path()
for thumbnail in os.listdir(thumbnail_path):
os.remove(os.path.join(thumbnail_path, thumbnail))
def thumbnail_name(data_filename):
"""Returns the name of the thumbnail corresponding to the specified data file."""
m = hashlib.md5(data_filename)
return os.path.join(pathfinder.thumbnails_path(), m.hexdigest() + '.png')
def test_thumbnail_path(self):
"""Verify correct path to data thumbnails"""
thumb_path = os.path.join(self.user_path, 'thumbnails')
self.assertEqual(thumb_path, pathfinder.thumbnails_path())
def remove_thumbs(self):
"""Removes all the thumnbail plots"""
thumbnail_path = pathfinder.thumbnails_path()
for thumbnail in os.listdir(thumbnail_path):
os.remove(os.path.join(thumbnail_path, thumbnail))
def thumbnail_name(data_filename):
"""Returns the name of the thumbnail corresponding to the specified data file."""
m = hashlib.md5(data_filename)
return os.path.join(pathfinder.thumbnails_path(), m.hexdigest() + '.png')
def test_thumbnail_path(self):
"""Verify correct path to data thumbnails"""
thumb_path = os.path.join(self.user_path, 'thumbnails')
self.assertEqual(thumb_path, pathfinder.thumbnails_path())
def test_thumbnail_name(self):
"""Verify thumbnail_name function returns the proper thumbnail filename"""
m = hashlib.md5(self.sample_data_file)
expected_thumb_name = os.path.join(pathfinder.thumbnails_path(), m.hexdigest() + '.png')
self.assertEqual(expected_thumb_name, model.thumbnail_name(self.sample_data_file))
