Update unittests.py