Fix #4. Additional unit test to reflect changes in #2

Odile@pc2050
1 parent 2a312522
Showing 2 changed files with 6 additions and 1 deletions Show diff stats
src/files.py
tests/test_files.py
@@ -37,7 +37,7 @@ class PireneaFiles(object):
         for path, _dirs, files in os.walk(self.folder):
             for filename in files:
                 if filename.startswith("P"):
-                    print("File %s has already a prefix.")
+                    print("File {0} has already a prefix.".format(filename))
                 else:
                     new_filename = prefix + filename
                     os.rename(os.path.join(path, filename), os.path.join(path, new_filename))
@@ -40,7 +40,12 @@ class PireneaFilesTestCase(unittest.TestCase):
         with tempfile.TemporaryDirectory(prefix="PIRENEA_DATA") as tmpdirname:
             with tempfile.TemporaryFile(dir=tmpdirname) as tf:
                 self.pirenea = PireneaFiles(tmpdirname)
+                # Check add prefix
                 self.pirenea.add_prefix("P0")
+                # Check if add prefix has no effect if already added
+                self.pirenea.add_prefix("P0")
+                self.pirenea.remove_prefix("P0")
+                # Check if remove prefix has no effect if file has no prefix
                 self.pirenea.remove_prefix("P0")
                 """Check if temporary file tf is unchanged"""
                 self.assertTrue(os.path.isfile(tf.name))