-
Notifications
You must be signed in to change notification settings - Fork 419
/
comment_utils_test.py
38 lines (28 loc) · 1.15 KB
/
comment_utils_test.py
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
28
29
30
31
32
33
34
35
36
37
38
import unittest
from sweepai.utils.comment_utils import check_comments_presence
class TestCommentUtils(unittest.TestCase):
def test_check_comments_presence(self):
# Test Python file with a comment
self.assertEqual(
check_comments_presence("test.py", "# This is a comment"), True
)
# Test Python file without a comment
self.assertEqual(
check_comments_presence("test.py", 'print("Hello, World!")'), False
)
# Test JavaScript file with a comment
self.assertEqual(
check_comments_presence("test.js", "// This is a comment"), True
)
# Test JavaScript file without a comment
self.assertEqual(
check_comments_presence("test.js", 'console.log("Hello, World!");'), False
)
# Test unsupported file type with a comment
self.assertEqual(
check_comments_presence("test.txt", "# This is a comment"), False
)
# Test unsupported file type without a comment
self.assertEqual(check_comments_presence("test.txt", "Hello, World!"), False)
if __name__ == "__main__":
unittest.main()