Unit test with Python