Advanced Software Engineering Homework
Typicall best practice is to put tests in a seperate test module, then use pytest to run all *_test.py files.
For easy grading, I put all test into it's respective task*.py file. Just use "python3 task*.py" to run the function and tests.