Update unittests for Python project