Skip to content

Commit fde6b72

Browse files
committed
Fixed type hint in unit test code
1 parent d14fbc1 commit fde6b72

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/test_argparse_completer.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
import argparse
77
import numbers
88
from typing import (
9+
Dict,
910
List,
1011
cast,
1112
)
@@ -285,12 +286,12 @@ def do_raise_completion_error(self, args: argparse.Namespace) -> None:
285286
############################################################################################################
286287
# Begin code related to receiving arg_tokens
287288
############################################################################################################
288-
def choices_takes_arg_tokens(self, arg_tokens: argparse.Namespace) -> List[str]:
289+
def choices_takes_arg_tokens(self, arg_tokens: Dict[str, List[str]]) -> List[str]:
289290
"""Choices function that receives arg_tokens from ArgparseCompleter"""
290291
return [arg_tokens['parent_arg'][0], arg_tokens['subcommand'][0]]
291292

292293
def completer_takes_arg_tokens(
293-
self, text: str, line: str, begidx: int, endidx: int, arg_tokens: argparse.Namespace
294+
self, text: str, line: str, begidx: int, endidx: int, arg_tokens: Dict[str, List[str]]
294295
) -> List[str]:
295296
"""Completer function that receives arg_tokens from ArgparseCompleter"""
296297
match_against = [arg_tokens['parent_arg'][0], arg_tokens['subcommand'][0]]

0 commit comments

Comments
 (0)