Amended testdd.py with assertEqual