Support python3
Use assertEqual instead of assertEquals in the tests (the latter is deprecated)