@@ -286,8 +286,8 @@ class MyClass:
286286 analyzer = CodeAnalyzer ()
287287 tree = ast .parse (code )
288288 analyzer .visit (tree )
289- assert len (analyzer .structure ) == 1
290- assert analyzer .structure [0 ]["class_name" ] == "MyClass"
289+ assert len (analyzer .analyzed_classes ) == 1
290+ assert analyzer .analyzed_classes [0 ]["class_name" ] == "MyClass"
291291
292292
293293def test_codeanalyzer_finds_multiple_classes ():
@@ -304,8 +304,8 @@ class ClassB:
304304 analyzer = CodeAnalyzer ()
305305 tree = ast .parse (code )
306306 analyzer .visit (tree )
307- assert len (analyzer .structure ) == 2
308- class_names = sorted ([c ["class_name" ] for c in analyzer .structure ])
307+ assert len (analyzer .analyzed_classes ) == 2
308+ class_names = sorted ([c ["class_name" ] for c in analyzer .analyzed_classes ])
309309 assert class_names == ["ClassA" , "ClassB" ]
310310
311311
@@ -320,9 +320,9 @@ def my_method(self):
320320 analyzer = CodeAnalyzer ()
321321 tree = ast .parse (code )
322322 analyzer .visit (tree )
323- assert len (analyzer .structure ) == 1
324- assert len (analyzer .structure [0 ]["methods" ]) == 1
325- assert analyzer .structure [0 ]["methods" ][0 ]["method_name" ] == "my_method"
323+ assert len (analyzer .analyzed_classes ) == 1
324+ assert len (analyzer .analyzed_classes [0 ]["methods" ]) == 1
325+ assert analyzer .analyzed_classes [0 ]["methods" ][0 ]["method_name" ] == "my_method"
326326
327327
328328def test_codeanalyzer_finds_multiple_methods ():
@@ -339,8 +339,8 @@ def method_b(self):
339339 analyzer = CodeAnalyzer ()
340340 tree = ast .parse (code )
341341 analyzer .visit (tree )
342- assert len (analyzer .structure ) == 1
343- method_names = sorted ([m ["method_name" ] for m in analyzer .structure [0 ]["methods" ]])
342+ assert len (analyzer .analyzed_classes ) == 1
343+ method_names = sorted ([m ["method_name" ] for m in analyzer .analyzed_classes [0 ]["methods" ]])
344344 assert method_names == ["method_a" , "method_b" ]
345345
346346
@@ -354,7 +354,7 @@ def top_level_function():
354354 analyzer = CodeAnalyzer ()
355355 tree = ast .parse (code )
356356 analyzer .visit (tree )
357- assert len (analyzer .structure ) == 0
357+ assert len (analyzer .analyzed_classes ) == 0
358358
359359
360360def test_codeanalyzer_class_with_no_methods ():
@@ -367,9 +367,9 @@ class MyClass:
367367 analyzer = CodeAnalyzer ()
368368 tree = ast .parse (code )
369369 analyzer .visit (tree )
370- assert len (analyzer .structure ) == 1
371- assert analyzer .structure [0 ]["class_name" ] == "MyClass"
372- assert len (analyzer .structure [0 ]["methods" ]) == 0
370+ assert len (analyzer .analyzed_classes ) == 1
371+ assert analyzer .analyzed_classes [0 ]["class_name" ] == "MyClass"
372+ assert len (analyzer .analyzed_classes [0 ]["methods" ]) == 0
373373
374374
375375# --- Test Data for Parameterization ---
@@ -489,10 +489,10 @@ class TestCodeAnalyzerArgsReturns:
489489 "code_snippet, expected_args, expected_return" , TYPE_TEST_CASES
490490 )
491491 def test_type_extraction (self , code_snippet , expected_args , expected_return ):
492- structure , imports , types = parse_code (code_snippet )
492+ analyzed_classes , imports , types = parse_code (code_snippet )
493493
494- assert len (structure ) == 1 , "Should parse one class"
495- class_info = structure [0 ]
494+ assert len (analyzed_classes ) == 1 , "Should parse one class"
495+ class_info = analyzed_classes [0 ]
496496 assert class_info ["class_name" ] == "TestClass"
497497
498498 assert len (class_info ["methods" ]) == 1 , "Should find one method"
0 commit comments