Function _assign

numexpr3/ne3compiler.py:898–914

AST function handler for an intermediate assignment in a statement block. The convention is that intermediate targets are named targets if they do not exist in the calling namespace and secondary outputs if they have been pre-allocated.

(self: NumExpr, node: ast.AST)

# Note: these use 'self', which must be a NumExpr object 
# ('self' is not a reserved keyword in Python)
def _assign(self: NumExpr, node: ast.AST) -> NumReg:
    '''
    AST function handler for an intermediate assignment in a statement block. 
    The convention is that intermediate targets are named targets if they do not 
    exist in the calling namespace and secondary outputs if they have been 
    pre-allocated.
    '''
    # print( '$ast.Assign' )
    # node.targets is a list; It must have a len=1 for NumExpr3 (i.e. no multiple returns)
    # node.value is likely a BinOp, Call, Comparison, or BoolOp
    if len(node.targets) != 1:
        raise ValueError('NumExpr3 supports only singleton returns in assignments.')
    
    valueReg = _ASTAssembler[type(node.value)](self, node.value)
    # Not populating self.assignTarget here, it's only needed for id'ing the 
    # output.
    return _mutate(self, node.targets[0], valueReg)

def _assign_last(self: NumExpr, node: ast.AST) -> NumReg:
    '''