Update tests-unit.py