19 lines
1.1 KiB
Python
19 lines
1.1 KiB
Python
|
import unittest
|
|||
|
|
from dop1 import parse
|
|||
|
|
|
|||
|
|
|
|||
|
|
class MyTestCase(unittest.TestCase):
|
|||
|
|
def test_dop1(self):
|
|||
|
|
self.assertEqual(parse('А ты знал, что ВТ - лучшая кафедра в ИТМО?'), ['ВТ лучшая кафедра в ИТМО'])
|
|||
|
|
self.assertEqual(parse('А ты знал, что ВТ - лучшая\nкафедра в ИТМО?'), ['ВТ лучшая кафедра в ИТМО'])
|
|||
|
|
self.assertEqual(parse('А ты знал, что ВТ - лучшая крутая супер пупер кафедра в ИТМО?'), [])
|
|||
|
|
self.assertEqual(parse('ВТ ИТМО ВТ ИТМО'), ['ВТ ИТМО ВТ ИТМО'])
|
|||
|
|
self.assertEqual(parse('ВТИТМО'), [])
|
|||
|
|
self.assertEqual(parse('ВТ.ИТМО'), [])
|
|||
|
|
self.assertEqual(parse('ВТ ИТМО'), ['ВТ ИТМО'])
|
|||
|
|
self.assertEqual(parse('А ты знал, что ВТ - лучшая кафедра в ИТМО?\nДа да, ВТ - самая крутая кафедра в ИТМО'), ['ВТ лучшая кафедра в ИТМО', 'ВТ самая крутая кафедра в ИТМО'])
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
unittest.main()
|