def test_correctly_placed_symbol(self, mypy_nodes: MypyNodeFactory) -> None: hand, gen = mypy_nodes.get_method() gen_map = mypy_nodes.get_generated_stubs_map() result = lookup_symbol(gen_map, hand) assert result == LookupResult(symbol=gen, containing_class=gen.info)
def test_mislocated_symbol_with_cls_not_found( self, mypy_nodes: MypyNodeFactory) -> None: mislocated_handwritten = mypy_nodes.get_mislocated_method_handwritten() gen_map = copy(mypy_nodes.get_generated_stubs_map()) del gen_map[mislocated_handwritten.info.fullname] result = lookup_symbol(gen_map, mislocated_handwritten) assert result == LookupResult(symbol=None, containing_class=None)
def test_mislocated_symbol(self, mypy_nodes: MypyNodeFactory) -> None: mislocated_handwritten = mypy_nodes.get_mislocated_method_handwritten() gen_map = mypy_nodes.get_generated_stubs_map() mislocated_actual_location = ( mypy_nodes.get_mislocated_method_actual_location_generated()) result = lookup_symbol(gen_map, mislocated_handwritten) assert result == LookupResult( symbol=mislocated_actual_location, containing_class=mislocated_actual_location.info, )
def test_mislocated_symbol_without_cls( self, mypy_nodes: MypyNodeFactory) -> None: mislocated_handwritten = mypy_nodes.get_mislocated_method_handwritten() gen_map = mypy_nodes.get_generated_stubs_map() original_func_info = mislocated_handwritten.info mislocated_handwritten.info = FUNC_NO_INFO result = lookup_symbol(gen_map, mislocated_handwritten) assert result == LookupResult(symbol=None, containing_class=None) mislocated_handwritten.info = original_func_info