I am working with a codebase that doesn’t have any tests setup. I am trying to add some tests and currently have the below test class test_main.py