def test_not_string_input(self):
try:
helper_functions.process_extension(35)
self.failIf(True,
"Non-string filename - ints can't split")
except AttributeError as ae :
self.assert_(True)
def test_file_in_period_named_directory(self):
""" A directory name that includes a period can lead to screwy results.
Make sure that doesn't happen again.
"""
test_filename = os.sep.join(['.first','second'])
found_extension = helper_functions.process_extension(test_filename)
self.assert_(found_extension is None)
def test_file_in_period_named_directory(self):
""" A directory name that includes a period can lead to screwy results.
Make sure that doesn't happen again.
"""
test_filename = os.sep.join(['.first', 'second'])
found_extension = helper_functions.process_extension(test_filename)
self.assert_(found_extension is None)
def process_filename(fqn, files_by_size, extensions):
""" Given a fully-qualified filename, a dictionary of filenames keyed by
size, and a dictionary of file extensions, update the two dictionaries
with the relevant information about the file.
"""
filesize = os.stat(fqn).st_size
helper_functions.add_or_append(filesize, fqn, files_by_size)
extension = helper_functions.process_extension(fqn)
if extension is None:
extension = NO_EXTENSION
helper_functions.add_or_append(extension, filesize, extensions)
def process_filename(fqn, files_by_size, extensions):
""" Given a fully-qualified filename, a dictionary of filenames keyed by
size, and a dictionary of file extensions, update the two dictionaries
with the relevant information about the file.
"""
filesize = os.stat(fqn).st_size
helper_functions.add_or_append(filesize, fqn, files_by_size)
extension = helper_functions.process_extension(fqn)
if extension is None:
extension = NO_EXTENSION
helper_functions.add_or_append(extension, filesize, extensions)
def test_normal_filenames(self):
test_extensions = ['txt', 'avi', 'JPEG', 'py', 'pyc', 'mov',
'wmv', 'mp3']
test_names = ['kitty', 'RENAME', 'personal', 'CamelCaseMe',
'weird_name_1', 'weird-unixy-name']
# Cartesian product filename list
test_filenames = [ '.'.join([name, ext])
for name in test_names
for ext in test_extensions ]
found_extensions = [ helper_functions.process_extension(test_filename)
for test_filename in test_filenames ]
self.assertEquals(set(found_extensions),
set(test_extensions),
"Missed or misfiled one or more extensions")
def test_normal_filenames(self):
test_extensions = [
'txt', 'avi', 'JPEG', 'py', 'pyc', 'mov', 'wmv', 'mp3'
]
test_names = [
'kitty', 'RENAME', 'personal', 'CamelCaseMe', 'weird_name_1',
'weird-unixy-name'
]
# Cartesian product filename list
test_filenames = [
'.'.join([name, ext]) for name in test_names
for ext in test_extensions
]
found_extensions = [
helper_functions.process_extension(test_filename)
for test_filename in test_filenames
]
self.assertEquals(set(found_extensions), set(test_extensions),
"Missed or misfiled one or more extensions")
def test_no_extension(self):
test_filename = 'I_am_myself'
found_extension = helper_functions.process_extension(test_filename)
self.assert_(found_extension is None)
def test_zero_length_extension(self):
test_filename = 'just_me.'
found_extension = helper_functions.process_extension(test_filename)
self.assertEquals('', found_extension)
def test_not_string_input(self):
try:
helper_functions.process_extension(35)
self.failIf(True, "Non-string filename - ints can't split")
except AttributeError as ae:
self.assert_(True)
def test_no_extension(self):
test_filename = 'I_am_myself'
found_extension = helper_functions.process_extension(test_filename)
self.assert_(found_extension is None)
def test_zero_length_extension(self):
test_filename = 'just_me.'
found_extension = helper_functions.process_extension(test_filename)
self.assertEquals('', found_extension)
