Exemplos de _broadcast em Python

Exemplo n.º 1

def test_broadcast_success():
    A = np.empty((5, 1))
    B = np.empty((1, 6))
    C = np.empty((6, ))
    D = np.empty((1, ))
    array_shapes = [arr.shape for arr in [A, B, C, D]]
    repl._broadcast(array_shapes)

Exemplo n.º 2

def test_broadcast_fail():
    A = np.empty((5, 1))
    B = np.empty((1, 6))
    C = np.empty((5, ))
    D = np.empty((1, ))
    array_shapes = [arr.shape for arr in [A, B, C, D]]
    try:
        repl._broadcast(array_shapes)
    except SyntaxError:
        return
    assert (False)