Change assertion to error

Author: p-sawicki

mypyc/irbuild/specialize.py

@@ -1025,10 +1025,10 @@ def translate_object_new(builder: IRBuilder, expr: CallExpr, callee: RefExpr) ->
             f"{call} not supported for classes inheriting from non-native classes", expr.line
         )
         return None
+    if len(expr.args) != 1:
+        builder.error(f"{call} supported only with 1 argument, got {len(expr.args)}", expr.line)
+        return None
 
-    assert (
-        len(expr.args) == 1
-    ), f"Expected object.__new__() call to have exactly 1 argument, got {len(expr.args)}"
     typ_arg = expr.args[0]
     method_args = fn.fitem.arg_names
     if isinstance(typ_arg, NameExpr) and len(method_args) > 0 and method_args[0] == typ_arg.name:

mypyc/test-data/irbuild-classes.test

@@ -2052,6 +2052,14 @@ class ClsListAssignment:
         [cls, val] = [object, "object"]  # E: Assignment to argument cls in __new__ method unsupported
         return object.__new__(cls)
 
+class WrongNumberOfArgs:
+    def __new__(cls):
+        return super().__new__()  # E: object.__new__() supported only with 1 argument, got 0
+
+class WrongNumberOfArgsObjectNew:
+    def __new__(cls):
+        return object.__new__(cls, 1)  # E: object.__new__() supported only with 1 argument, got 2
+
 [case testClassWithFreeList]
 from mypy_extensions import mypyc_attr, trait