Added python 3 support (hopefully)
Fix: "DeprecationWarning: Please use assertEqual instead".