Update test cases for case:
        self.assertTrue(output == expected)
to
        self.assertEquals(expected, output )
Update test cases for case:
        self.assertTrue(output == expected)
to
        self.assertEquals(expected, output )