Deleting a file using python

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

Here is what the above code is Doing:
1. First, it checks if the file exists.
2. If it exists, it removes the file.
3. It then removes the empty directory.
4. Finally, it removes the non-empty directory and its content.