def test_current_directory(self): """ Ensure that calling the function with no arguments returns the contents of the current working directory """ expected = {".doc": 1, ".txt": 2} extensions_count = examine_directory.examine(path=self.tempdirectory) self.assertEqual(extensions_count, expected)
def test_empty_directory(self): """ Ensure that calling the function on an empty directory returns nothing / empty list """ self.emptydirectory = tempfile.mkdtemp("emptydir") expected = {} extension_count = examine_directory.examine(path=self.emptydirectory) self.assertEqual(extension_count, expected) #cleanup shutil.rmtree(self.emptydirectory) print("Deleted ", self.emptydirectory)