Show an error if the class has non-annotated attributes and we want t…

…o convert it to a Mojo struct

py2mojo/converters/classdef.py

@@ -4,6 +4,7 @@
 
 from tokenize_rt import Token
 
+from ..exceptions import ParseException
 from ..helpers import ast_to_offset, find_token
 from ..rules import RuleSet
 
@@ -16,6 +17,10 @@ def _replace_class_keyword(tokens: list, i: int, rules: RuleSet) -> None:
 def convert_classdef(node: ast.FunctionDef, rules: RuleSet) -> Iterable:
     """Converts the annotation of the given function definition."""
     if rules.convert_class_to_struct:
+        for child in node.body:
+            if isinstance(child, ast.Assign):
+                raise ParseException(node, 'Class contains non type annotated attributes.')
+
         offset = ast_to_offset(node)
         yield (
             offset,

tests/test_classdef.py

@@ -1,4 +1,7 @@
+import pytest
+
 from helpers import validate
+from py2mojo.exceptions import ParseException
 from py2mojo.rules import RuleSet
 
 
@@ -12,3 +15,17 @@ def test_classdef():
         'struct Foo(Bar): pass',
         rules=RuleSet(convert_class_to_struct=True),
     )
+
+
+def test_classdef_non_fully_annotated_functions():
+    validate(
+        '''class Number: number = 10''',
+        '''class Number: number = 10''',
+        rules=RuleSet(convert_class_to_struct=False),
+    )
+    with pytest.raises(ParseException):
+        validate(
+            '''class Number: number = 10''',
+            '''class Number: number = 10''',
+            rules=RuleSet(convert_class_to_struct=True),
+        )