import os import shutil if os.path.exists("demofile.txt"): os.remove("demofile.txt") # one file at a time os.rmdir("test_directory") # removes empty directory shutil.rmtree("test_directory") # removes not empty directory and its content