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