Added documentation to tests.py file for assign2