def test_1(self): sol = Solution() lst = 'ab' res = sol.get_res(lst) self.assertItemsEqual( res, ['AB', 'Ab', 'aB', 'ab'] )
def test_2(self): sol = Solution() lst = 'abc' res = sol.get_res(lst) self.assertItemsEqual( res, ['ABC', 'ABc', 'AbC', 'Abc', 'aBC', 'aBc', 'abC', 'abc'] )