add tests for function expend_annotations

Author: KsanaKozlova

numba_typing/tests/test_type_annotations.py

@@ -1,17 +1,17 @@
 import unittest
 import type_annotations
-from typing import Union, Dict, List
+from typing import Union, Dict, List, TypeVar
 
 
 class TestTypeAnnotations(unittest.TestCase):
 
-    def test_get_func_annotations_exception(self):
+    def test_get_func_annotations_exceptions(self):
 
         def foo(a: int, b, c: str = "string"):
             pass
         with self.assertRaises(SyntaxError) as raises:
             type_annotations.get_func_annotations(foo)
-        self.assertIn('Not found annotation for parameter b', str(raises.exception))
+        self.assertIn('No annotation for parameter b', str(raises.exception))
 
     def test_get_cls_annotations(self):
         class TestClass(object):
@@ -46,6 +46,17 @@ def func_three(a: Dict[int, str], b: str = "string", c: int = 1):
             with self.subTest(func=f.__name__):
                 self.assertEqual(type_annotations.get_func_annotations(f), expected)
 
+    def test_expend_annotations(self):
+
+        S = TypeVar('S', float, str)
+        info = ({'a': [int], 'b': [int, float], 'c': [S]}, {})
+        result = type_annotations.expend_annotations(info)
+        expected = [[{'a': int, 'b': int, 'c': float}, {}],
+                    [{'a': int, 'b': int, 'c': str}, {}],
+                    [{'a': int, 'b': float, 'c': float}, {}],
+                    [{'a': int, 'b': float, 'c': str}, {}]]
+        self.assertEqual(result, expected)
+
 
 if __name__ == '__main__':
     unittest.main()

numba_typing/type_annotations.py

@@ -1,5 +1,5 @@
 from inspect import signature
-from typing import get_type_hints, Union
+from typing import get_type_hints, Union, TypeVar
 from itertools import product
 from copy import deepcopy