Add check for missing location info on an AST node

This matches the behavior of ast.get_source_segment().

Author: lieryan

rope/base/codeanalyze.py

@@ -199,6 +199,8 @@ def get_source_segment(self, node, *, padded: bool = False) -> str | None:
 
     def __getitem__(self, node) -> tuple[int, int] | tuple[None, None]:
         try:
+            if node.end_lineno is None or node.end_col_offset is None:
+                return (None, None)
             line_idx = node.lineno - 1
             end_line_idx = node.end_lineno - 1
             col_offset = node.col_offset

ropetest/codeanalyzetest.py

@@ -1250,20 +1250,40 @@ def test_padding_length_is_char_count_not_byte_count(self, dict_node, ast_adapte
     class TestMissingLocationInfoTest:
         SOURCE = 'abc = def'
 
-        def test_get_source_segment_returns_none(self, ast_adapter):
+        def test_get_source_segment_returns_none_without_location(self, ast_adapter):
             node = ast.Module(body=[], type_ignores=[])
             assert not hasattr(node, 'lineno')
             assert not hasattr(node, 'col_offset')
 
             assert ast_adapter.get_source_segment(node) is None
 
-        def test_get_source_region_returns_none(self, ast_adapter):
+        def test_get_source_region_returns_none_without_location(self, ast_adapter):
             node = ast.Module(body=[], type_ignores=[])
             assert not hasattr(node, 'lineno')
             assert not hasattr(node, 'col_offset')
 
             assert ast_adapter[node] == (None, None)
 
+        def test_get_source_segment_returns_none_without_end_location(self, ast_adapter):
+            node = ast.Add(
+                lineno=1,
+                col_offset=1,
+                end_lineno=None,
+                end_col_offset=None,
+            )
+
+            assert ast_adapter.get_source_segment(node) is None
+
+        def test_get_source_region_returns_none_without_end_location(self, ast_adapter):
+            node = ast.Add(
+                lineno=1,
+                col_offset=1,
+                end_lineno=None,
+                end_col_offset=None,
+            )
+
+            assert ast_adapter[node] == (None, None)
+
     class TestRegionOffset:
         SOURCE = textwrap.dedent(
             """