Added assert statement.

diff --git a/Python/compile.c b/Python/compile.c
index 3bdd994ea39804c42abb712abd0e6a89866cadcf..0338d6ff45f37569d913fc4c26b392695c9b5e41 100644 (file)
--- a/Python/compile.c
+++ b/Python/compile.c
@@ -1925,6 +1925,48 @@ com_expr_stmt(c, n)
        }
 }
 
+static void
+com_assert_stmt(c, n)
+       struct compiling *c;
+       node *n;
+{
+       int a = 0, b = 0;
+       int i;
+       REQ(n, assert_stmt); /* 'assert' test [',' test] */
+       /* Generate code like for
+          
+          if __debug__:
+             if not <test>:
+                raise AssertionError [, <message>]
+
+          where <message> is the second test, if present.
+       */
+       if (Py_OptimizeFlag)
+               return;
+       com_addopnamestr(c, LOAD_GLOBAL, "__debug__");
+       com_push(c, 1);
+       com_addfwref(c, JUMP_IF_FALSE, &a);
+       com_addbyte(c, POP_TOP);
+       com_pop(c, 1);
+       com_node(c, CHILD(n, 1));
+       com_addfwref(c, JUMP_IF_TRUE, &b);
+       com_addbyte(c, POP_TOP);
+       com_pop(c, 1);
+       /* Raise that exception! */
+       com_addopnamestr(c, LOAD_GLOBAL, "AssertionError");
+       com_push(c, 1);
+       i = NCH(n)/2; /* Either 2 or 4 */
+       if (i > 1)
+               com_node(c, CHILD(n, 3));
+       com_addoparg(c, RAISE_VARARGS, i);
+       com_pop(c, i);
+       /* The interpreter does not fall through */
+       /* All jumps converge here */
+       com_backpatch(c, a);
+       com_backpatch(c, b);
+       com_addbyte(c, POP_TOP);
+}
+
 static void
 com_print_stmt(c, n)
        struct compiling *c;
@@ -2866,6 +2908,9 @@ com_node(c, n)
        case exec_stmt:
                com_exec_stmt(c, n);
                break;
+       case assert_stmt:
+               com_assert_stmt(c, n);
+               break;
        case if_stmt:
                com_if_stmt(c, n);
                break;