Update tests.py

Problem #9 and #7's unit tests - added two tests for each of the problems