18 lines
1.1 KiB
Python
18 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()
|