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