37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
import unittest
|
|
|
|
from crypto_square import encode
|
|
|
|
|
|
# Tests adapted from `problem-specifications//canonical-data.json` @ v3.2.0
|
|
|
|
class CryptoSquareTest(unittest.TestCase):
|
|
def test_empty_string(self):
|
|
self.assertEqual(encode(''), '')
|
|
|
|
def test_lowercase(self):
|
|
self.assertEqual(encode('A'), 'a')
|
|
|
|
def test_remove_spaces(self):
|
|
self.assertEqual(encode(' b '), 'b')
|
|
|
|
def test_remove_punctuation(self):
|
|
self.assertEqual(encode('@1,%!'), '1')
|
|
|
|
def test_9chars_results_3chunks(self):
|
|
self.assertEqual(encode('This is fun!'), 'tsf hiu isn')
|
|
|
|
def test_8chars_results_3chunks_ending_space(self):
|
|
self.assertEqual(encode('Chill out.'), 'clu hlt io ')
|
|
|
|
def test_54chars_results_7chunks_2ending_space(self):
|
|
self.assertEqual(
|
|
encode('If man was meant to stay on the ground, '
|
|
'god would have given us roots.'),
|
|
'imtgdvs fearwer mayoogo anouuio ntnnlvt wttddes aohghn sseoau '
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|