示例#1
0
def test_length():
    assert length(5) == 1
    assert length(5.) == 1
    assert length([1, 2, 3]) == 3

    assert length(np.array(5)) == 1
    assert length(np.array([5])) == 1
    assert length(np.array([1, 2, 3])) == 3
    assert length(np.ones((3, 2))) == 3

    assert length(cas.GenMX_ones(5)) == 5
def test_casadi_list():
    assert is_casadi_type(
        [cas.GenMX_ones(5), cas.GenMX_ones(5)], recursive=False) == False
    assert is_casadi_type(
        [cas.GenMX_ones(5), cas.GenMX_ones(5)], recursive=True) == True
def test_mixed_list():
    assert is_casadi_type([np.array(5), cas.GenMX_ones(5)],
                          recursive=False) == False
    assert is_casadi_type([np.array(5), cas.GenMX_ones(5)],
                          recursive=True) == True
def test_casadi():
    assert is_casadi_type(cas.GenMX_ones(5), recursive=False) == True
    assert is_casadi_type(cas.GenMX_ones(5), recursive=True) == True