Fixed test functions to work by using the proper assertEquals function from the unittest library rather than the python assert function