Python lab: add unit tests for stats.py