import unittest class TestFindDiff(unittest.TestCase): def test_find_diff(self): solution = Solution() self.assertRaises(TypeError, solution.find_diff, None) self.assertEqual(solution.find_diff('ab', 'aab'), 'a') self.assertEqual(solution.find_diff('aab', 'ab'), 'a') self.assertEqual(solution.find_diff('abcd', 'abcde'), 'e') self.assertEqual(solution.find_diff('aaabbcdd', 'abdbacade'), 'e') self.assertEqual(solution.find_diff_xor('ab', 'aab'), 'a') self.assertEqual(solution.find_diff_xor('aab', 'ab'), 'a') self.assertEqual(solution.find_diff_xor('abcd', 'abcde'), 'e') self.assertEqual(solution.find_diff_xor('aaabbcdd', 'abdbacade'), 'e') print('Success: test_find_diff') def main(): test = TestFindDiff() test.test_find_diff() if __name__ == '__main__': main()