Fix an edge case with multiples param being the first one "self"

msaelices · Sep 10, 2023 · ac770fd · ac770fd
1 parent 69cd1c1
commit ac770fd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/py2mojo/converters/functiondef.py b/py2mojo/converters/functiondef.py
@@ -42,7 +42,7 @@ def convert_functiondef(node: ast.FunctionDef, level: int = 0) -> Iterable:
         new_type = get_mojo_type(curr_type)
 
         if not new_type:
-            return
+            continue
 
         yield (
             ast_to_offset(arg),

diff --git a/tests/test_functiondef.py b/tests/test_functiondef.py
@@ -39,3 +39,14 @@ def test_functiondef_with_list_types():
         'def concat(l1: list, l2: list) -> list: return l1 + l2',  # no changed
     )
 
+def test_functiondef_inside_classes():
+    validate(
+'''
+class Point:
+    def __init__(self, x: int, y: int) -> int: ...
+''',
+'''
+class Point:
+    def __init__(self, x: Int, y: Int) -> Int: ...
+''',
+    )