def testGetFileExtension(self): extensions = ["x", "xy", "xyz"] partial_filename = "dummy" for ext in extensions: this_ext = ut.getFileExtension("%s.%s" % (partial_filename, ext)) self.assertEqual(this_ext, ext) # Try with no extension this_ext = ut.getFileExtension("%s" % partial_filename) self.assertIsNone(this_ext)
""" Utility to change the extension of a file """ from util import changeFileExtension, getFileExtension import os CUR_EXT = "sci" NEW_EXT = "scish" files = [ff for ff in os.listdir(".") \ if getFileExtension(ff) == CUR_EXT] for ff in files: new_file = changeFileExtension(ff, NEW_EXT) os.rename(ff, new_file)