diff --git a/src/RestrictedPython/transformer.py b/src/RestrictedPython/transformer.py index c030bb7..7934761 100644 --- a/src/RestrictedPython/transformer.py +++ b/src/RestrictedPython/transformer.py @@ -356,7 +356,7 @@ def gen_unpack_wrapper(self, node, target): return (tmp_target, cleanup) def gen_none_node(self): - return ast.NameConstant(value=None) + return ast.Constant(None) def gen_del_stmt(self, name_to_del): return ast.Delete(targets=[ast.Name(name_to_del, ast.Del())]) diff --git a/tests/transformer/test_gen.py b/tests/transformer/test_gen.py new file mode 100644 index 0000000..9165d2a --- /dev/null +++ b/tests/transformer/test_gen.py @@ -0,0 +1,9 @@ +import ast + +from RestrictedPython.transformer import RestrictingNodeTransformer + + +def test_RestrictingNodeTransformer__gen_none_node__1(): + node = RestrictingNodeTransformer().gen_none_node() + assert node.value is None + assert isinstance(node, ast.Constant)