def test_1_row(self): str_ = 'PAYPALISHIRINGIN' str_converted = zig_zag_conversion(str_, 1) self.assertEqual(str_converted, 'PAYPALISHIRINGIN')
def test_3_rows(self): str_ = 'PAYPALISHIRINGIN' str_converted = zig_zag_conversion(str_, 3) self.assertEqual(str_converted, 'PAHNAPLSIIGNYIRI')
def test_4_rows(self): str_ = 'NEWENGLANDPATRIOTS' str_converted = zig_zag_conversion(str_, 4) self.assertEqual(str_converted, 'NLTEGAARSWNNPITEDO')
def test_2_row(self): str_ = 'COMPUTERS' str_converted = zig_zag_conversion(str_, 2) self.assertEqual(str_converted, 'CMUESOPTR')