Coverage for src/renaissance/recipes/simplify_renaissance.py: 100%

1"""AI: Recipe that simplifies Renaissance's own unit-to-pytest conversion files.""" 

2 

3from pathlib import Path 

4from typing import override 

5 

6from renaissance.recipes.python_refactoring import PythonRefactoring 

7 

8 

9class SimplifyRenaissance(PythonRefactoring): 

10 """AI: Recipe that simplifies Renaissance's own unit-to-pytest conversion files.""" 

11 

12 def __init__(self, file): 

13 """AI: Prepare a refactoring processor scoped to Renaissance's own unit-to-pytest conversion files.""" 

14 super().__init__(file) 

15 self.white_list_pattern = "unit_to_pytest" 

16 self.black_list_pattern = "SimplifyRenaissance" 

17 

18 @override 

19 def run(self): 

20 if self.black_list_pattern in self.filename or self.white_list_pattern not in self.filename: 

21 print(f"skipping: {Path(self.filename).resolve()}") 

22 return 

23 

24 print(f"simplify {Path(self.filename).resolve()}") 

25 self.replace_stmt("$val = match.expansions[$key][0].signature", "$val= match[$key]") 

26 self.replace_stmt( 

27 "factory = ASTFactory(PythonASTNode)\n$atu = factory.create_from_text($code, $name)", 

28 "PythonASTNode.load_from_text($code, $name)", 

29 ) 

30 self.commit()