Change assertTrue(a == b) in unittest to assertEquals.