88
99class TestRemoveSignature (unittest .TestCase ):
1010 @parameterized .expand ([ # type: ignore[misc]
11- ("multiple_newlines" ,
11+ (
12+ "multiple_newlines" ,
1213 "Could you please review the attached document?\n \n \n I need your feedback by Friday." ,
13- "Could you please review the attached document? I need your feedback by Friday."
14+ "Could you please review the attached document? I need your feedback by Friday." ,
1415 ),
1516 (
1617 "multiple_spaces" ,
1718 "The meeting is scheduled for 3PM tomorrow." ,
18- "The meeting is scheduled for 3PM tomorrow."
19+ "The meeting is scheduled for 3PM tomorrow." ,
1920 ),
2021 (
2122 "mixed_whitespace" ,
2223 "Please find the report attached. \n \n The numbers look good \r \n \r \n for Q3!" ,
23- "Please find the report attached. The numbers look good for Q3!"
24+ "Please find the report attached. The numbers look good for Q3!" ,
2425 ),
2526 (
2627 "empty_string" ,
@@ -31,7 +32,7 @@ class TestRemoveSignature(unittest.TestCase):
3132 "trailing_whitespace" ,
3233 "I'll send the updated version tomorrow. \n \n " ,
3334 "I'll send the updated version tomorrow."
34- ),
35+ )
3536 ])
3637 def test_remove_newline (self , name : str , input_text : str , expected : str ) -> None :
3738 self .assertEqual (remove_newline (input_text ), expected )
@@ -81,12 +82,11 @@ def test_remove_newline(self, name: str, input_text: str, expected: str) -> None
8182 "no_greetings" ,
8283 "This is a plain text without any greetings or signatures." ,
8384 "This is a plain text without any greetings or signatures."
84- )
85+ ),
8586 ])
8687 def test_remove_greetings_and_following_text (self , name : str , input_text : str , expected : str ) -> None :
8788 self .assertEqual (remove_greetings_and_following_text (input_text ), expected )
8889
8990
90- if __name__ == '__main__' :
91- unittest .main ()
91+ if __name__ == "__main__" :
9292 unittest .main ()
0 commit comments