To fix
DeprecationWarning: Please use assertEqual instead.

Change-Id: Iae7cb50dd968d055b106e35baa7eee526a50bf6d