Use assertEqual instead of assertEquals in tests since it's deprecated in Python 3