Python Unit Test Assertequal