src/test.py: fix tests