1import unittest
 2
 3from solution import *
 4
 5
 6class TestSanity(unittest.TestCase):
 7    """Sanity test for organize."""
 8
 9    def test_sanity(self):
10        """Ensure name exists and is callable.."""
11        self.assertTrue(callable(parse_wishlist))
12
13
14if __name__ == '__main__':
15    unittest.main()

  1import unittest
  2
  3# Thanks to this guy we need to prevent hacking. -> https://py-fmi.org/student/75
  4anti_rusalov_false = unittest.TestCase.assertFalse
  5anti_rusalov_true = unittest.TestCase.assertTrue
  6anti_rusalov_is = unittest.TestCase.assertIs
  7
  8from solution import *
  9
 10modified_example = '''(3.14) Иван Иванов [10 г.]
 11
 12 - Плейстейшан
 13 - Количка с дистанционо
 14 * Братче
 15
 16(1.94)   Georgi "Jorkata" Georgiev  [43]
 17
 18 -
 19  - Vancheto ot tretiq etaj
 20'''
 21half = '''(3.14) Иван Иванов [10 г.]
 22
 23 - Плейстейшан
 24 - Количка с дистанционо
 25 * Братче
 26
 27(1.94)   Georgi "Jorkata" Georgiev  [43]
 28
 29 -
 30  - Vancheto ot tretiq etaj
 31
 32
 33
 34(99.9534412345) На мама сладкото ангелче [3 годинки]
 35    * Най-хубавото дървено конче (размер 1.5)
 36    * Най-страхотното лего (тамън отвориха лицензиран магазин)
 37    * Хлороформ
 38    *
 39
 40
 41'''
 42full = '''(3.14) Иван Иванов [10 г.]
 43
 44 - Плейстейшан
 45 - Количка с дистанционо
 46 * Братче
 47
 48(1.94)   Georgi "Jorkata" Georgiev  [43]
 49
 50 -
 51  - Vancheto ot tretiq etaj
 52
 53( 6.9 ) Mr. D-r [ 999 ]
 54 - [Doctor stuff] Скалпел
 55 - [Doctor stuff] Мехлем за дупе
 56 - [Non-doctor stuff] Мехлем за дупе
 57
 58
 59(1) Gospodin (he/him) Mitko [12  год.]
 60
 61
 62 - 1
 63 -2
 64 - 3
 65 - 4
 66 - Malko mi pisva da go pisha toq test
 67
 68(99.9534412345) На мама сладкото ангелче [3 годинки]
 69    * Най-хубавото дървено конче (размер 1.5)
 70    * Най-страхотното лего (тамън отвориха лицензиран магазин)
 71    * Хлороформ
 72    *
 73
 74
 75'''
 76
 77
 78expected_example = [(3.14, 'Иван Иванов', 10, ('Плейстейшан', 'Количка с дистанционо', 'Братче')),
 79                    (1.94, 'Georgi "Jorkata" Georgiev', 43, ('Vancheto ot tretiq etaj', ))]
 80expected_half = [(3.14, 'Иван Иванов', 10, ('Плейстейшан', 'Количка с дистанционо', 'Братче')),
 81                 (1.94, 'Georgi "Jorkata" Georgiev', 43, ('Vancheto ot tretiq etaj', )),
 82                 (99.9534412345, 'На мама сладкото ангелче', 3, ('Най-хубавото дървено конче (размер 1.5)', 'Най-страхотното лего (тамън отвориха лицензиран магазин)', 'Хлороформ'))]
 83expected_full = [(3.14, 'Иван Иванов', 10, ('Плейстейшан', 'Количка с дистанционо', 'Братче')),
 84                 (1.94, 'Georgi "Jorkata" Georgiev', 43, ('Vancheto ot tretiq etaj', )),
 85                 (6.9, 'Mr. D-r', 999, ('[Doctor stuff] Скалпел', '[Doctor stuff] Мехлем за дупе', '[Non-doctor stuff] Мехлем за дупе')),
 86                 (1, 'Gospodin (he/him) Mitko', 12, ('1', '2', '3', '4', 'Malko mi pisva da go pisha toq test')),
 87                 (99.9534412345, 'На мама сладкото ангелче', 3, ('Най-хубавото дървено конче (размер 1.5)', 'Най-страхотното лего (тамън отвориха лицензиран магазин)', 'Хлороформ'))]
 88class TestRegex(unittest.TestCase):
 89    """Test for parse_wishlist."""
 90
 91    @classmethod
 92    def setUpClass(cls):
 93        cls.maxDiff = None
 94
 95    def test_modified_example(self):
 96        """A test similar to the example (just a lil' bit different)."""
 97        try:
 98            # Converted first, because of otherwise misleading output
 99            converted = [tuple(map(str, element[:3])) + (element[3],)
100                         for element in expected_example]
101            self.assertEqual(converted, parse_wishlist(modified_example))
102        except AssertionError:
103            self.assertEqual(expected_example, parse_wishlist(modified_example))
104
105    def test_half(self):
106        """A test without the more edgy cases."""
107        try:
108            # Converted first, because of otherwise misleading output
109            converted = [tuple(map(str, element[:3])) + (element[3],)
110                         for element in expected_example]
111            self.assertEqual(converted, parse_wishlist(modified_example))
112        except AssertionError:
113            self.assertEqual(expected_half, parse_wishlist(half))
114
115    def test_full(self):
116        """A single test to rule them all."""
117        try:
118            # Converted first, because of otherwise misleading output
119            converted = [tuple(map(str, element[:3])) + (element[3],)
120                         for element in expected_example]
121            self.assertEqual(converted, parse_wishlist(modified_example))
122        except AssertionError:
123            self.assertEqual(expected_full, parse_wishlist(full))
124
125
126if __name__ == '__main__':
127    unittest.main()