def test_get_type_deep(self): self.assertEqual(List[List[int]], get_type([[1, 2], [3]])) self.assertEqual(List[List[object]], get_type([[1, 2], ['3']])) deep = [[[1, 2]], [[3]], [[4], [5, '6']]] self.assertEqual(List[List[List[object]]], get_type(deep)) self.assertEqual(Dict[str, List[List[List[object]]]], get_type({'a': deep}))
def test_get_type_union(self): union_type = get_type(Union[int, str]) # The following line should not raise. Union[(union_type, )] self.assertTrue(instance_of(Union[int, str], union_type))
def _store_cls_info(result: object, original_obj: dict, kwargs): if kwargs.get('_store_cls', None) and isinstance(result, dict): cls = get_type(original_obj) if cls.__module__ == 'typing': cls_name = repr(cls) else: cls_name = get_class_name(cls, fully_qualified=True) result['-cls'] = cls_name
def _store_cls_info(result: object, attr: str, original_obj: dict, **kwargs): if isinstance(result, dict) and kwargs.get('_store_cls'): cls = get_type(original_obj[attr]) if cls.__module__ == 'typing': cls_name = repr(cls) else: cls_name = get_class_name(cls, fully_qualified=True, fork_inst=kwargs['fork_inst']) result['-cls'] = cls_name
def test_get_type_callable(self): def func1(x: int, y: str) -> object: pass def func2() -> int: pass def func3(x: int): pass def func4(): pass def func5(x): pass def func6(l: List[List[List[int]]]): pass self.assertEqual(Callable[[int, str], object], get_type((func1))) self.assertEqual(Callable[[], int], get_type(func2)) self.assertEqual(Callable[[int], NoneType], get_type(func3)) self.assertEqual(Callable, get_type(func4)) self.assertEqual(Callable[[Any], NoneType], get_type(func5)) self.assertEqual(Callable[[List[List[List[int]]]], NoneType], get_type(func6))
def test_get_type_of_ndarray(self): arr_type = get_type(numpy.array([1, 2, 3])) self.assertEqual(numpy.ndarray, arr_type)
def _determine_cls(obj: Iterable, cls: Optional[type]) -> Optional[type]: cls_ = cls if not cls and hasattr(obj, '__getitem__') and len(obj) > 0: obj_with_only_one_elem = obj.__getitem__(slice(0, 1)) cls_ = get_type(obj_with_only_one_elem) return cls_
def test_get_type_set(self): self.assertEqual(Set[int], get_type({1, 2, 3})) self.assertEqual(Set[object], get_type({1, 2, '3'})) self.assertEqual(Set[Unknown], get_type(set([])))
def test_get_type_list(self): self.assertEqual(List[int], get_type([1, 2, 3])) self.assertEqual(List[object], get_type([1, 2, '3'])) self.assertEqual(List[Unknown], get_type([]))
def test_get_type_any(self): any_type = get_type(Any) self.assertEqual(Any, any_type)
def test_get_type_some_class(self): class SomeRandomClass: pass self.assertEqual(SomeRandomClass, get_type(SomeRandomClass()))
def test_get_type_type(self): self.assertEqual(Type[int], get_type(int))
def test_get_type_list_with_union(self): self.assertEqual(List[Union[int, str, float]], get_type([1, '2', 3.0], True))
def test_get_type_lambda(self): self.assertEqual(Callable[[Unknown, Unknown], Unknown], get_type(lambda x, y: 42)) self.assertEqual(Callable[[], Unknown], get_type(lambda: 42))
def test_get_type_async(self): async def func(x: int) -> str: return '42' self.assertEqual(Callable[[int], Awaitable[str]], get_type(func))
def test_get_type_tuple(self): self.assertEqual(Tuple[int, int, int], get_type((1, 2, 3))) self.assertEqual(Tuple[int, str, str], get_type((1, '2', '3')))
def test_get_type_dict(self): self.assertEqual(Dict[str, int], get_type({'a': 1, 'b': 2})) self.assertEqual(Dict[int, object], get_type({1: 'a', 2: 2})) self.assertEqual(Dict[Unknown, Unknown], get_type({}))