Change the def keyword to fn if the function has no arguments with --…

…level=1

py2mojo/converters/functiondef.py

@@ -39,9 +39,6 @@ def _add_declaration(tokens: list, i: int, level: int, declaration: str) -> None
 
 def convert_functiondef(node: ast.FunctionDef, level: int = 0) -> Iterable:
  """Converts the annotation of the given function definition."""
- if not node.args.args:
- return
-
  if level > 0:
  offset = ast_to_offset(node)
  yield (
@@ -51,6 +48,9 @@ def convert_functiondef(node: ast.FunctionDef, level: int = 0) -> Iterable:
  ),
  )
 
+ if not node.args.args:
+ return
+
  for arg in node.args.args:
  if arg.arg == 'self':
  yield (

tests/test_functiondef.py

@@ -2,13 +2,25 @@
 
 from helpers import validate
 
-# parametrize the converted types
+
+def test_functiondef_with_no_params():
+ validate(
+ 'def main(): print("Hello world!")',
+ 'def main(): print("Hello world!")',
+ )
+ validate(
+ 'def main(): print("Hello world!")',
+ 'fn main(): print("Hello world!")',
+ level=1,
+ )
+
+
 @pytest.mark.parametrize(
  'python_type, mojo_type',
  [
  ('int', 'Int'),
  ('float', 'Float64'),
- ]
+ ],
 )
 def test_functiondef_with_basic_types(python_type, mojo_type):
  validate(
@@ -27,7 +39,7 @@ def test_functiondef_with_basic_types(python_type, mojo_type):
  [
  ('int', 'Int'),
  ('float', 'Float64'),
- ]
+ ],
 )
 def test_functiondef_with_list_types(python_type, mojo_type):
  validate(
@@ -47,13 +59,14 @@ def test_functiondef_with_list_types(python_type, mojo_type):
  '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__(inout self, x: Int, y: Int) -> Int: ...
 ''',