#16 Add list default value pattern.

Author: simon-ritchie

numdoclint/helper.py

@@ -849,13 +849,18 @@ def _remove_type_bracket_block_from_args_str(args_str: str):
         return args_str
     bracket_count: int = 0
     result_str: str = ''
+    is_type_block: bool = False
     for char in args_str:
-        if char == '[':
+        if bracket_count == 0 and char == ':':
+            is_type_block = True
+        if is_type_block and char == '[':
             bracket_count += 1
             continue
-        if char == ']':
+        if is_type_block and char == ']':
             bracket_count -= 1
             continue
+        if bracket_count == 0 and (char == ',' or char == '='):
+            is_type_block = False
         if bracket_count != 0:
             continue
         result_str += char

tests/test_helper.py

@@ -1809,3 +1809,14 @@ def test__remove_type_bracket_block_from_args_str():
         'dict_val: Optional = None, tuple_val: Optional=None'
     )
     assert result_str == expected_str
+
+    args_str: str = (
+        'list_val_1: List[int],'
+        ' list_val_2: Optional[List[int]] = [100, 200]'
+    )
+    result_str = helper._remove_type_bracket_block_from_args_str(
+        args_str=args_str)
+    expected_str = (
+        'list_val_1: List, list_val_2: Optional = [100, 200]'
+    )
+    assert result_str == expected_str