Skip to content

Commit eb6b61c

Browse files
authored
Update test_generate_analyzer.py
renamed variable.
1 parent 75e9ecf commit eb6b61c

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

scripts/microgenerator/tests/unit/test_generate_analyzer.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -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

293293
def 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

328328
def 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

360360
def 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

Comments
 (0)