diff --git a/src/pystencilssfg/ir/entities.py b/src/pystencilssfg/ir/entities.py index d97e81022e611c9e3ac87ba14e69a8f6ebdfb46b..8d798250490748f4c2e50c9a11b867ca66c9f0e8 100644 --- a/src/pystencilssfg/ir/entities.py +++ b/src/pystencilssfg/ir/entities.py @@ -343,6 +343,8 @@ class SfgMethod(SfgClassMember, CommonFunctionProperties): ): super().__init__(cls) + self._name = name + from .postprocessing import CallTreePostProcessing param_collector = CallTreePostProcessing() @@ -363,6 +365,10 @@ class SfgMethod(SfgClassMember, CommonFunctionProperties): attributes, ) + @property + def name(self) -> str: + return self._name + @property def static(self) -> bool: return self._static diff --git a/tests/generator_scripts/source/Conditionals.py b/tests/generator_scripts/source/Conditionals.py index 9016b73744f78fef504bb4f09b6742a630a6b12d..6c2490808f9617e6780b1ffcf94af614d254fd02 100644 --- a/tests/generator_scripts/source/Conditionals.py +++ b/tests/generator_scripts/source/Conditionals.py @@ -22,7 +22,7 @@ with SourceFileGenerator() as sfg: ) ) - sfg.function("getRating", "int32")( + sfg.function("getRating").returns("int32")( sfg.switch(noodle, autobreak=False) .case("Noodles::RIGATONI")( "return 13;" @@ -39,7 +39,7 @@ with SourceFileGenerator() as sfg: "return 0;" ) - sfg.function("isItalian", return_type="bool")( + sfg.function("isItalian").returns("bool")( sfg.branch( sfg.expr("{0} == Noodles::RIGATONI || {0} == Noodles::SPAGHETTI", noodle) )(