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')