Exemplo n.º 1
0
def test_dim_handling():
    assert dim_handling([x], dim=2) == {x: (False, False)}
    assert dim_handling([x, y], dims={
        x: 1,
        y: 2
    }) == {
        x: (False, True),
        y: (False, False)
    }
    assert dim_handling([x], broadcastables={x: (False, )}) == {x: (False, )}
Exemplo n.º 2
0
def test_dim_handling():
    assert dim_handling([x], dim=2) == {x: (False, False)}
    assert dim_handling([x, y], dims={x: 1, y: 2}) == {x: (False, True),
                                                       y: (False, False)}
    assert dim_handling([x], broadcastables={x: (False,)}) == {x: (False,)}