How to write test using pyunit in Python