commit 9509f44d79b8b18c43922b72ed5e139919120050
author Werner Lewis <werner.lewis@arm.com> Wed Aug 24 17:46:22 2022 +0100
committer Werner Lewis <werner.lewis@arm.com> Wed Sep 21 16:46:39 2022 +0100
tree 1b2dc113fcaec5aaf911749eb636dae8a2a82775
parent 6d04142972989e64b7172be62b54b7e9fb6fc3b6

Add missing typing

Signed-off-by: Werner Lewis <werner.lewis@arm.com>

tests/scripts/generate_bignum_tests.py

diff --git a/tests/scripts/generate_bignum_tests.py b/tests/scripts/generate_bignum_tests.py
index 9551e21..016e037 100755
--- a/tests/scripts/generate_bignum_tests.py
+++ b/tests/scripts/generate_bignum_tests.py
@@ -49,7 +49,7 @@
 import sys
 
 from abc import ABCMeta, abstractmethod
-from typing import Callable, Dict, Iterator, List, Optional, Tuple, TypeVar
+from typing import Callable, Dict, Iterator, List, Tuple, TypeVar
 
 import scripts_path # pylint: disable=unused-import
 from mbedtls_dev import test_case
@@ -57,10 +57,10 @@
 
 T = TypeVar('T') #pylint: disable=invalid-name
 
-def hex_to_int(val):
+def hex_to_int(val: str) -> int:
     return int(val, 16) if val else 0
 
-def quote_str(val):
+def quote_str(val) -> str:
     return "\"{}\"".format(val)
 
 
@@ -89,7 +89,7 @@
         "0000000000000000123", "-0000000000000000123",
         "1230000000000000000", "-1230000000000000000"
     ] # type: List[str]
-    input_cases = [] # type: List[Tuple[str, ...]]
+    input_cases = [] # type: List[Tuple[str, str]]
 
     def __init__(self, val_l: str, val_r: str) -> None:
         self.arg_l = val_l
@@ -97,10 +97,10 @@
         self.int_l = hex_to_int(val_l)
         self.int_r = hex_to_int(val_r)
 
-    def arguments(self):
+    def arguments(self) -> List[str]:
         return [quote_str(self.arg_l), quote_str(self.arg_r), self.result()]
 
-    def description(self):
+    def description(self) -> str:
         """Generate a description for the test case.
 
         If not set, case_description uses the form A `symbol` B, where symbol
@@ -148,7 +148,7 @@
         return tmp
 
     @classmethod
-    def get_value_pairs(cls) -> Iterator[Tuple[str, ...]]:
+    def get_value_pairs(cls) -> Iterator[Tuple[str, str]]:
         """Generator for pairs of inputs.
 
         Combinations are first generated from all input values, and then
@@ -176,12 +176,12 @@
         ("2b5", "2b6")
         ]
 
-    def __init__(self, val_l, val_r):
+    def __init__(self, val_l, val_r) -> None:
         super().__init__(val_l, val_r)
         self._result = int(self.int_l > self.int_r) - int(self.int_l < self.int_r)
         self.symbol = ["<", "==", ">"][self._result + 1]
 
-    def result(self):
+    def result(self) -> str:
         return str(self._result)
 
 
@@ -191,7 +191,7 @@
     test_function = "mbedtls_mpi_cmp_abs"
     test_name = "MPI compare (abs)"
 
-    def __init__(self, val_l, val_r):
+    def __init__(self, val_l, val_r) -> None:
         super().__init__(val_l.strip("-"), val_r.strip("-"))
 
 
@@ -207,11 +207,11 @@
         ], 2
     ))
 
-    def __init__(self, val_l, val_r):
+    def __init__(self, val_l, val_r) -> None:
         super().__init__(val_l, val_r)
         self.symbol = "+"
 
-    def result(self):
+    def result(self) -> str:
         return quote_str(hex(self.int_l + self.int_r).replace("0x", "", 1))