File size: 1,025 Bytes
127bfda
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import unittest

import bot.byte_semantics_reply as reply_logic


class TestByteSemanticsReplyExtra(unittest.TestCase):
    def test_split_text_edge_cases(self):
        # Empty text
        self.assertEqual(reply_logic.split_text_for_chat(""), [])
        self.assertEqual(reply_logic.split_text_for_chat(None), [])

        # Long text no spaces
        self.assertEqual(len(reply_logic.split_text_for_chat("a" * 1000, max_len=400)), 2)

    def test_extract_movie_title_variations(self):
        # Movie title with "que estamos assistindo" (should return empty)
        self.assertEqual(reply_logic.extract_movie_title("ficha tecnica do que estamos vendo"), "")

        # Movie title with punctuation
        self.assertEqual(
            reply_logic.extract_movie_title("Ficha tecnica do filme Matrix!!!"), "Matrix"
        )

        # Generic terms
        self.assertEqual(reply_logic.extract_movie_title("ficha tecnica hoje"), "")
        self.assertEqual(reply_logic.extract_movie_title("ficha tecnica agora"), "")