#
#     This file is part of CasADi.
#
#     CasADi -- A symbolic framework for dynamic optimization.
#     Copyright (C) 2010-2023 Joel Andersson, Joris Gillis, Moritz Diehl,
#                             KU Leuven. All rights reserved.
#     Copyright (C) 2011-2014 Greg Horn
#
#     CasADi is free software; you can redistribute it and/or
#     modify it under the terms of the GNU Lesser General Public
#     License as published by the Free Software Foundation; either
#     version 3 of the License, or (at your option) any later version.
#
#     CasADi is distributed in the hope that it will be useful,
#     but WITHOUT ANY WARRANTY; without even the implied warranty of
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
#     Lesser General Public License for more details.
#
#     You should have received a copy of the GNU Lesser General Public
#     License along with CasADi; if not, write to the Free Software
#     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
#
#
from casadi import *
import casadi as c
import numpy
import unittest
from types import *
from helpers import *
import pickle
import os
import sys
from casadi.tools import capture_stdout

scipy_interpolate = False
try:
  import scipy.interpolate
  scipy.interpolate.RectBivariateSpline
  scipy_interpolate = True
except:
  pass


class Functiontests(casadiTestCase):

  def test_call_empty(self):
    x = SX.sym("x",2)
    fsx = Function("fsx", [x,[]],[x])
    x = MX.sym("x",2)
    fmx1 = Function("fmx1", [x,MX()],[x])
    fmx2 = Function("fmx2", [x,[]],[x])

    for f in [fsx,fmx1,fmx2]:
      f(0,0)

      X = MX.sym("X",2)
      F = f(X,MX())
      g = Function("g", [X],[F])

      g(0)

    x = SX.sym("x",2)
    fsx = Function("fsx", [x],[x,[]])
    x = MX.sym("x",2)
    fmx1 = Function("fmx1", [x],[x,MX()])
    fmx2 = Function("fmx2", [x],[x,[]])

    for f in [fsx,fmx1,]:
      f(0)

      X = MX.sym("X",2)
      F = f(X)
      g = Function("g", [X],F)

      g(0)

  def test_MX_funSeed(self):
    self.message("MX_funSeed")
    x1 = MX.sym("x",2)
    y1 = MX.sym("y")
    x2 = MX.sym("x",2)
    y2 = MX.sym("y")
    p= Function("p", [x1,y1,x2,y2],[sin(x1) + y1,sin(x2) + y2])

    n1 = DM([4,5])
    N1 = 3
    n2 = DM([5,7])
    N2 = 8

    out = p(n1,N1,n2,N2)

    self.checkarray(sin(n1)+N1,out[0],"output")
    self.checkarray(sin(n2)+N2,out[1],"output")
  def test_segfault(self):
    f = Function()
    with self.assertRaises(Exception):
      f.stats()
  def test_issue304(self):
    self.message("regression test for #304") # this code used to segfault
    x = SX.sym("x")

    f = Function("f", [x],[x**2,x**3])

    X = MX.sym("X")

    z=f(X)

    g = Function("g", [X], z).expand()

  def test_jacobian(self):
    x = SX.sym("x",3,1)
    y = SX.sym("y",2,1)

    f = Function("f", [x,y],[x**2,y,x*y[0]])

    g = jacobian_old(f, 0, 0)

    self.assertEqual(g.n_in(),f.n_in())
    self.assertEqual(g.n_out(),f.n_out()+1)

  def test_xfunction(self):
    x = SX.sym("x",3,1)
    y = SX.sym("y",2,1)

    f = Function("f", [x,y],[x**2,y,x*y[0]])

    X = MX.sym("x",3,1)
    Y = MX.sym("y",2,1)

    F = Function("F", [X,Y],[X**2,Y,X*Y[0]])

    self.checkfunction(f,F,inputs=[[0.1,0.7,1.3],[7.1,2.9]],sens_der=False,evals=False)


  @memory_heavy()
  def test_jacobians(self):

    x = SX.sym("x")

    self.assertEqual(jacobian(5,x).nnz(),0)


    def test(sp):
      x = SX.sym("x",sp.size2())
      sp2 = jacobian(mtimes(DM.ones(sp),x),x).sparsity()
      self.checkarray(sp.row(),sp2.row());
      self.checkarray(sp.colind(),sp2.colind());

    for i in range(5):
      test(Sparsity.lower(i))
      test(Sparsity.lower(i).T)
      test(Sparsity.dense(i,i))
      test(Sparsity.diag(i))

    for i in [63,64,65,127,128,129]:
      d = Sparsity.diag(i)
      test(d)

      test(d + Sparsity.rowcol([0],[5],i,i))

      b = Sparsity.band(i,-1) + Sparsity.band(i,1)
      test(b + Sparsity.rowcol([0],[5],i,i))

    m = DM.ones(Sparsity.diag(129))
    m[:50,0] = 1
    m[60:,0] = 1
    m[6:9,6] = 1
    m[9,9:12] = 1

    sp = m[:,:120].sparsity()

    test(sp)
    #test(sp.T)

    m = DM.ones(Sparsity.diag(64))
    m[:50,0] = 1
    m[60:,0] = 1

    sp = m.T[:40,:].sparsity()
    test(sp)
    test(sp.T)

    sp = m[:40,:].sparsity()
    test(sp)
    test(sp.T)

    sp = m.T[:20,:].sparsity()
    test(sp)
    test(sp.T)

    sp = m[:20,:].sparsity()
    test(sp)
    test(sp.T)

    for i in [63,64,65,127,128,129]:
      test(Sparsity.lower(i))
      test(Sparsity.lower(i).T)

    for n in ([63,64,65,127,128,129] if args.run_slow else [63,64,65]):
      for m in ([63,64,65,127,128,129] if args.run_slow else [63,64,65]):
        print((n,m))
        sp = Sparsity.dense(n,m)

        test(sp)

        random.seed(0)

        I = DM.ones(sp)
        for i in range(n):
          for j in range(m):
            if random.random()<0.5:
              I[i,j] = 0
        I = sparsify(I)

        sp_holes = I.sparsity()

        test(sp_holes)

        z = DM(sp_holes.size1(), sp_holes.size2())

        R = 5
        v = []
        for r in range(R):
          h = [z]*5
          h[r] = I
          v.append(horzcat(*h))
        d = vertcat(*v)

        test(d.sparsity())

  @memory_heavy()
  def test_hessians(self):
    def test(sp):
      x = SX.sym("x",sp.size2())
      self.assertTrue(sp==sp.T)
      f = Function("f", [x],[mtimes([x.T,DM.ones(sp),x])])
      J = hessian_old(f, 0, 0)
      sp2 = J.sparsity_out(0)
      self.checkarray(sp.row(),sp2.row())
      self.checkarray(sp.colind(),sp2.colind())

    A = DM([[1,1,0,0,0,0],[1,1,1,0,1,1],[0,1,1,1,0,0],[0,0,1,1,0,1],[0,1,0,0,1,0],[0,1,0,1,0,1]])
    A = sparsify(A)
    C = A.sparsity()

    test(C)

    A = DM([[1,0,0,0,0,0],[0,1,1,0,1,1],[0,1,1,1,0,0],[0,0,1,1,0,1],[0,1,0,0,1,0],[0,1,0,1,0,1]])
    A = sparsify(A)
    C = A.sparsity()

    test(C)

    A = DM([[1,0,0,0,0,0],[0,1,0,0,1,1],[0,0,1,1,0,0],[0,0,1,1,0,1],[0,1,0,0,1,0],[0,1,0,1,0,1]])
    A = sparsify(A)
    C = A.sparsity()

    test(C)

    A = DM([[0,0,0,0,0,0],[0,1,0,0,1,1],[0,0,1,1,0,0],[0,0,1,1,0,1],[0,1,0,0,1,0],[0,1,0,1,0,1]])
    A = sparsify(A)
    C = A.sparsity()

    test(C)

    A = DM([[0,0,0,0,0,0],[0,1,0,0,1,0],[0,0,1,1,0,0],[0,0,1,1,0,1],[0,1,0,0,1,0],[0,0,0,1,0,1]])
    A = sparsify(A)
    C = A.sparsity()

    test(C)


    for i in [63,64,65,100,127,128,129]:
      d = Sparsity.diag(i)
      test(d)

      test(d + Sparsity.rowcol([0],[5],i,i) + Sparsity.rowcol([5],[0],i,i))

      b = Sparsity.band(i,-1) + Sparsity.band(i,1)
      test(b)
      test(b + Sparsity.rowcol([0],[5],i,i) + Sparsity.rowcol([5],[0],i,i))

      d = Sparsity.dense(i,i)
      test(d)

      d = Sparsity.diag(i) + Sparsity.triplet(i,i,[0]*i,list(range(i)))+Sparsity.triplet(i,i,list(range(i)),[0]*i)
      test(d)


      sp = Sparsity.dense(i,i)

      random.seed(0)

      I = DM.ones(sp)
      for ii in range(i):
        for jj in range(i):
          if random.random()<0.5:
            I[ii,jj] = 0
            I[jj,ii] = 0
      I = sparsify(I)

      sp_holes = I.sparsity()

      test(sp_holes)

      z = DM(sp_holes.size1(), sp_holes.size2())

      R = 5
      v = []
      for r in range(R):
        h = [z]*5
        h[r] = I
        v.append(horzcat(*h))
      d = vertcat(*v)

      test(d.sparsity())

  def test_customIO(self):
    x = SX.sym("x")
    f = Function('f',[x],[x*x, x],["i0"], ["foo","bar"])

    ret = f(i0=12)

    self.checkarray(DM([144]),ret["foo"])
    self.checkarray(DM([12]),ret["bar"])


    with self.assertRaises(Exception):
      f_out["baz"]

    ret = f(i0=SX(12))
    self.checkarray(ret["foo"],DM([144]))
    self.checkarray(ret["bar"],DM([12]))
    with self.assertRaises(Exception):
      self.checkarray(ret["baz"],DM([12]))

  def test_derivative_simplifications(self):

    n = 1
    x = SX.sym("x",n)

    M = Function("M", [x],[mtimes((x-DM(list(range(n)))),x.T)])

    P = MX.sym("P",n,n)
    X = MX.sym("X",n)

    M_X= M(X)

    Pf = Function("P", [X, P], [mtimes(M_X,P)])

    P_P = jacobian_old(Pf, 1, 0)

    self.assertFalse("derivative" in str(P_P))

  def test_issue1464(self):
    n = 6
    x = SX.sym("x",n)
    u = SX.sym("u")


    N = 9

    rk4 = Function("f",[x,u],[x+u])

    for XX,XFunction in [(SX,Function),(MX,Function)]:

      g = []
      g2 = []


      V = XX.sym("V",(N+1)*n+N)
      VX,VU = vertsplit(V,[0,(N+1)*n,(N+1)*n+N])

      VXk = vertsplit(VX,n)
      VUk = vertsplit(VU,1)

      for k in range(N):

          xf = rk4(VXk[k],VUk[k])

          xfp = vertsplit(xf,int(n/2))
          vp = vertsplit(VXk[k+1],int(n/2))

          g.append(xfp[0] - vp[0])
          g.append(xfp[1] - vp[1])

          g2.append(xf-VXk[k+1])

      for i in range(2):
        f = XFunction("nlp",[V],[vertcat(*g)],{"ad_weight_sp":i})

        assert f.jac_sparsity(0, 0).nnz()==162

        f2 = XFunction("nlp",[V],[vertcat(*g2)],{"ad_weight_sp":i})

        assert f2.jac_sparsity(0, 0).nnz()==162

  def test_callback(self):
    class mycallback(Callback):
      def __init__(self, name, opts={}):
        Callback.__init__(self)
        self.construct(name, opts)

      def eval(self,argin):
        return [argin[0]**2]

    foo = mycallback("my_f")

    x = MX.sym('x')
    y = foo(x)

    f = Function("f",[x],[y])

    out = f(5)

    self.checkarray(out,25)

  def test_callback_buffer(self):
    class mycallback(Callback):
      def __init__(self, name, opts={}):
        Callback.__init__(self)
        self.construct(name, opts)
      def has_eval_buffer(self): return True
      def eval_buffer(self, arg, res):
        a = np.frombuffer(arg[0], dtype=np.float64)
        r = np.frombuffer(res[0], dtype=np.float64)
        r[:] = a**2
        return 0

    foo = mycallback("my_f")

    x = MX.sym('x')
    y = foo(x)

    f = Function("f",[x],[y])

    out = f(5)

    self.checkarray(out,25)

    class mycallback(Callback):
      def __init__(self, name, opts={}):
        Callback.__init__(self)
        self.construct(name, opts)
      def has_eval_buffer(self): return True
      def get_n_in(self): return 3
      def get_n_out(self): return 2
      def get_sparsity_in(self, i):
        if i==0:
          return Sparsity.dense(1,1)
        elif i==1:
          return Sparsity.dense(3,1)
        elif i==2:
          return Sparsity.dense(3,3)
      def get_sparsity_out(self, i):
        if i==0:
          return Sparsity.dense(3,1)
        elif i==1:
          return Sparsity.dense(3,3)
      def eval_buffer(self, arg, res):
        a = np.frombuffer(arg[0], dtype=np.float64)
        b = np.frombuffer(arg[1], dtype=np.float64)
        c = np.frombuffer(arg[2], dtype=np.float64).reshape((3,3), order='F')
        print(c)
        r0 = np.frombuffer(res[0], dtype=np.float64)
        r1 = np.frombuffer(res[1], dtype=np.float64).reshape((3,3), order='F')
        r0[:] = np.dot(a*c,b)
        r1[:,:] = c**2
        return 0

    foo = mycallback("my_f")

    a = 3
    b = DM([1,2,3])
    c = DM([[1,2,3],[4,5,6],[7,8,9]])

    res = foo(a,b,c)

    self.checkarray(res[0],mtimes(a*c,b))
    self.checkarray(res[1],c**2)

  def test_callback_errors(self):
    class mycallback(Callback):
      def __init__(self, name, opts={}):
        Callback.__init__(self)
        self.construct(name, opts)
      def eval(self,argin):
        raise Exception("foobar")

    foo = mycallback("my_f")

    x = MX.sym('x')
    y = foo(x)

    f = Function("f",[x],[y])

    try:
      f(3)
    except Exception as e:
      self.assertTrue("foobar" in str(e))

  def test_mapdict(self):
    x = SX.sym("x")
    y = SX.sym("y",2)
    z = SX.sym("z",2,2)
    v = SX.sym("z",Sparsity.upper(3))

    fun = Function("f",{"x":x,"y":y,"z":z,"v":v,"I":mtimes(z,y)+x,"II":sin(y*x).T,"III":v/x},["x","y","z","v"],["I","II","III"])

    n = 2

    X = [MX.sym("x") for i in range(n)]
    Y = [MX.sym("y",2) for i in range(n)]
    Z = [MX.sym("z",2,2) for i in range(n)]
    V = [MX.sym("z",Sparsity.upper(3)) for i in range(n)]

    res = fun.map(n).call({"x":horzcat(*X),"y":horzcat(*Y),"z":horzcat(*Z),"v":horzcat(*V)})

    res2 = fun.map(n).call([horzcat(*X),horzcat(*Y),horzcat(*Z),horzcat(*V)])

    F = Function("F",X+Y+Z+V,res2)
    F2 = Function("F",X+Y+Z+V,[res["I"],res["II"],res["III"]])

    np.random.seed(0)
    X_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in X ]
    Y_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in Y ]
    Z_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in Z ]
    V_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in V ]

    self.checkfunction(F,F2,inputs=X_+Y_+Z_+V_,jacobian=False,hessian=False,evals=False)

  @memory_heavy()
  def test_map_node(self):
    x = SX.sym("x")
    y = SX.sym("y",2)
    z = SX.sym("z",2,2)
    v = SX.sym("z",Sparsity.upper(3))

    fun = Function("f",[x,y,z,v],[mtimes(z,y)+x,sin(y*x).T,v/x])

    n = 2

    X = [MX.sym("x") for i in range(n)]
    Y = [MX.sym("y",2) for i in range(n)]
    Z = [MX.sym("z",2,2) for i in range(n)]
    V = [MX.sym("z",Sparsity.upper(3)) for i in range(n)]

    for parallelization in ["serial","openmp","unroll","inline","thread"] if args.run_slow else ["serial"]:
        print(parallelization)
        res = fun.map(n, parallelization).call([horzcat(*x) for x in [X,Y,Z,V]])


        F = Function("F",X+Y+Z+V,list(map(sin,res)))

        resref = [[] for i in range(fun.n_out())]
        for r in zip(X,Y,Z,V):
          for i,e in enumerate(map(sin,fun.call(r))):
            resref[i] = resref[i] + [e]

        Fref = Function("F",X+Y+Z+V,[horzcat(*x) for x in resref])

        np.random.seed(0)
        X_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in X ]
        Y_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in Y ]
        Z_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in Z ]
        V_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in V ]

        for f in [F, F.expand('expand_'+F.name())]:

          self.checkfunction(f,Fref,inputs=X_+Y_+Z_+V_,sparsity_mod=args.run_slow)

  def test_map_node_light(self):
    x = SX.sym("x")
    y = SX.sym("y",2)
    z = SX.sym("z",2,2)
    v = SX.sym("z",Sparsity.upper(3))

    fun = Function("f",[x,y,z,v],[mtimes(z,y)+x,sin(y*x).T,v/x])

    n = 2

    X = [MX.sym("x") for i in range(n)]
    Y = [MX.sym("y",2) for i in range(n)]
    Z = [MX.sym("z",2,2) for i in range(n)]
    V = [MX.sym("z",Sparsity.upper(3)) for i in range(n)]

    for parallelization in ["serial","openmp","unroll","inline","thread"]:
        print(parallelization)
        res = fun.map(n, parallelization).call([horzcat(*x) for x in [X,Y,Z,V]])


        F = Function("F",X+Y+Z+V,list(map(sin,res)))

        resref = [[] for i in range(fun.n_out())]
        for r in zip(X,Y,Z,V):
          for i,e in enumerate(map(sin,fun.call(r))):
            resref[i] = resref[i] + [e]

        Fref = Function("F",X+Y+Z+V,[horzcat(*x) for x in resref])

        np.random.seed(0)
        X_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in X ]
        Y_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in Y ]
        Z_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in Z ]
        V_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in V ]

        for f in [F, F.expand('expand_'+F.name())]:
          self.checkfunction_light(f,Fref,inputs=X_+Y_+Z_+V_,)

  def test_map_node_n_threads(self):
    x = SX.sym("x")
    y = SX.sym("y",2)
    z = SX.sym("z",2,2)
    v = SX.sym("z",Sparsity.upper(3))

    fun = Function("f",[x,y,z,v],[mtimes(z,y)+x,sin(y*x).T,v/x])

    X_ = [ DM(x.sparsity(),np.random.random(x.nnz())) for i in range(10) ]
    Y_ = [ DM(y.sparsity(),np.random.random(y.nnz())) for i in range(10) ]
    Z_ = [ DM(z.sparsity(),np.random.random(z.nnz())) for i in range(10) ]
    V_ = [ DM(v.sparsity(),np.random.random(v.nnz())) for i in range(10) ]


    print(fun.map(3,"thread",2))


    self.checkfunction_light(fun.map(2,"thread",1),fun.map(2),inputs=[hcat(X_[:2]),hcat(Y_[:2]),hcat(Z_[:2]),hcat(V_[:2])])
    self.checkfunction_light(fun.map(3,"thread",1),fun.map(3),inputs=[hcat(X_[:3]),hcat(Y_[:3]),hcat(Z_[:3]),hcat(V_[:3])])
    self.checkfunction_light(fun.map(3,"thread",2),fun.map(3),inputs=[hcat(X_[:3]),hcat(Y_[:3]),hcat(Z_[:3]),hcat(V_[:3])])
    self.checkfunction_light(fun.map(4,"thread",2),fun.map(4),inputs=[hcat(X_[:4]),hcat(Y_[:4]),hcat(Z_[:4]),hcat(V_[:4])])
    self.checkfunction_light(fun.map(4,"thread",5),fun.map(4),inputs=[hcat(X_[:4]),hcat(Y_[:4]),hcat(Z_[:4]),hcat(V_[:4])])

  @memory_heavy()
  def test_mapsum(self):
    x = SX.sym("x")
    y = SX.sym("y",2)
    z = SX.sym("z",2,2)
    v = SX.sym("z",Sparsity.upper(3))

    fun = Function("f",[x,y,z,v],[mtimes(z,y)+x,sin(y*x).T,v/x])

    n = 2

    X = [MX.sym("x") for i in range(n)]
    Y = [MX.sym("y",2) for i in range(n)]
    Z = [MX.sym("z",2,2) for i in range(n)]
    V = [MX.sym("z",Sparsity.upper(3)) for i in range(n)]

    zi = 0
    for Z_alt in [Z,[MX()]*3]:
      zi+= 1
      for parallelization in ["serial","openmp","unroll","thread"]:
        res = fun.mapsum([horzcat(*x) for x in [X,Y,Z_alt,V]],parallelization) # Joris - clean alternative for this?

        for ad_weight_sp in [0,1]:
          F = Function("F",X+Y+Z+V,list(map(sin,res)),{"ad_weight": 0,"ad_weight_sp":ad_weight_sp})

          resref = [0 for i in range(fun.n_out())]
          for r in zip(X,Y,Z_alt,V):
            for i,e in enumerate(fun.call(r)):
              resref[i] = resref[i] + e

          Fref = Function("F",X+Y+Z+V,list(map(sin,resref)))

          np.random.seed(0)
          X_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in X ]
          Y_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in Y ]
          Z_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in Z ]
          V_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in V ]

          inputs = X_+Y_+Z_+V_

          if parallelization!="thread":
            self.check_codegen(F,inputs=inputs)

          for f in [F,toSX_fun(F)]:
            self.checkfunction(f,Fref,inputs=inputs,sparsity_mod=args.run_slow)


  @memory_heavy()
  def test_mapsum2(self):
    x = SX.sym("x")
    y = SX.sym("y",2)
    z = SX.sym("z",2,2)
    v = SX.sym("z",Sparsity.upper(3))

    fun = Function("f",[x,y,z,v],[mtimes(z,y)+x,sin(y*x).T,v/x])

    n = 2

    X = [MX.sym("x") for i in range(n)]
    Y = [MX.sym("y",2) for i in range(n)]
    Z = MX.sym("z",2,2)
    V = MX.sym("z",Sparsity.upper(3))

    for Z_alt in [Z]:

      for parallelization in ["true_map_sum","serial","openmp","unroll","thread"]:
        for ad_weight_sp in [0,1]:
          for ad_weight in [0,1]:

            if parallelization=="true_map_sum":
              F = fun.map(n,[False,False,True,True],[True,False,False],{"ad_weight_sp":ad_weight_sp,"ad_weight":ad_weight})
            else:
              F = fun.map("map",parallelization,n,[2,3],[0],{"ad_weight_sp":ad_weight_sp,"ad_weight":ad_weight})

            resref = [0 for i in range(fun.n_out())]
            acc = 0
            bl = []
            cl = []
            for r in zip(X,Y,[Z_alt]*n,[V]*n):
              a,b,c= fun(*r)
              acc = acc + a
              bl.append(b)
              cl.append(c)

            Fref = Function("F",[horzcat(*X),horzcat(*Y),Z,V],[acc,horzcat(*bl),horzcat(*cl)])

            np.random.seed(0)
            X_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in X ]
            Y_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in Y ]
            Z_ = DM(Z.sparsity(),np.random.random(Z.nnz()))
            V_ = DM(V.sparsity(),np.random.random(V.nnz()))

            inputs = [horzcat(*X_),horzcat(*Y_),Z_,V_]

            if parallelization!="thread":
              self.check_codegen(F,inputs=inputs)

            for f in [F,toSX_fun(F)]:
              self.checkfunction(f,Fref,inputs=inputs,sparsity_mod=args.run_slow)

            self.check_serialize(F,inputs=inputs)

  def test_repmatnode(self):
    x = MX.sym("x",2)

    y = sin(repmat(x**2,1,3))

    z = MX.sym("y",2,2)

    F = Function("f",[x,z],[sum2(sum1(y))])

    x = SX.sym("x",2)

    y = sin(repmat(x**2,1,3))
    z = SX.sym("y",2,2)

    Fref = Function("f",[x,z],[sum2(sum1(y))])

    x0 = DM([1,7])
    x1 = DM([[3,0],[2,4]])

    self.check_codegen(F,inputs=[x0,x1])
    self.checkfunction(F,Fref,inputs=[x0,x1])

  def test_repsumnode(self):

    x = MX.sym("x",2)
    z = MX.sym("y",2,2)

    F = Function("f",[x,z],[sin(repsum((x**2).T,1,2)),(cos(x**2)*2*x).T])

    x = SX.sym("x",2)
    z = SX.sym("y",2,2)


    Fref = Function("f",[x,z],[sin(repsum((x**2).T,1,2)),(cos(x**2)*2*x).T])

    x0 = DM([1,7])
    x1 = DM([[3,0],[2,4]])

    self.check_codegen(F,inputs=[x0,x1])
    self.checkfunction(F,Fref,inputs=[x0,x1])

  def test_unknown_options(self):
    x = SX.sym("x")

    with self.assertRaises(Exception):
      f = SXFunction("f", [x],[x],{"fooo": False})

    with self.assertRaises(Exception):
      f = SXFunction("f", [x],[x],{"ad_weight": "foo"})

    if not has_nlpsol("ipopt"):
      return

  @requires_nlpsol("ipopt")
  def test_unknown_options_stringvector(self):
    x = SX.sym("x")
    solver = nlpsol("mysolver", "ipopt", {"x":x,"f":x**2}, {"monitor": ["nlp_f"]})
    with capture_stdout() as result:
      solver = nlpsol("mysolver", "ipopt", {"x":x,"f":x**2}, {"monitor": ["abc"]})
    self.assertTrue("Ignoring monitor 'abc'. Available functions: nlp_f" in result[1])

  @memory_heavy()
  def test_mapaccum(self):

    x = SX.sym("x",2)
    y = SX.sym("y")
    z = SX.sym("z",2,2)
    v = SX.sym("v",Sparsity.upper(3))

    fun = Function("f",[x,y,z,v],[mtimes(z,x)+y,sin(y*x).T,v/y])

    n = 2

    X = MX.sym("x",x.sparsity())
    Y = [MX.sym("y",y.sparsity()) for i in range(n)]
    Z = [MX.sym("z",z.sparsity()) for i in range(n)]
    V = [MX.sym("v",v.sparsity()) for i in range(n)]

    np.random.seed(0)
    X_ = DM(x.sparsity(),np.random.random(x.nnz()))
    Y_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in Y ]
    Z_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in Z ]
    V_ = [ DM(i.sparsity(),np.random.random(i.nnz())) for i in V ]

    for ad_weight in range(2):
      for ad_weight_sp in range(2):
        F = fun.mapaccum("map",n,[0],[0],{"ad_weight_sp":ad_weight_sp,"ad_weight": ad_weight})

        F.forward(2)

        XP = X

        Y0s = []
        Y1s = []
        Xps = []
        for k in range(n):
          XP, Y0,Y1 = fun(XP,Y[k],Z[k],V[k])
          Y0s.append(Y0)
          Y1s.append(Y1)
          Xps.append(XP)
        Fref = Function("f",[X,horzcat(*Y),horzcat(*Z),horzcat(*V)],[horzcat(*Xps),horzcat(*Y0s),horzcat(*Y1s)])
        inputs = [X_,horzcat(*Y_),horzcat(*Z_),horzcat(*V_)]

        for f in [F,toSX_fun(F)]:

          self.checkfunction(f,Fref,inputs=inputs)
          self.check_codegen(f,inputs=inputs)

    fun = Function("f",[y,x,z,v],[mtimes(z,x)+y+c.trace(v)**2,sin(y*x).T,v/y])

    for ad_weight in range(2):
      for ad_weight_sp in range(2):
        F = fun.mapaccum("map",n,[1,3],[0,2],{"ad_weight_sp":ad_weight_sp,"ad_weight": ad_weight})

        XP = X
        VP = V[0]

        Y0s = []
        Y1s = []
        Xps = []
        Vps = []
        for k in range(n):
          XP, Y0, VP = fun(Y[k],XP,Z[k],VP)
          Y0s.append(Y0)
          Xps.append(XP)
          Vps.append(VP)

        Fref = Function("f",[horzcat(*Y),X,horzcat(*Z),V[0]],[horzcat(*Xps),horzcat(*Y0s),horzcat(*Vps)])
        inputs = [horzcat(*Y_),X_,horzcat(*Z_),V_[0]]

        for f in [F,toSX_fun(F)]:
          self.checkfunction(f,Fref,inputs=inputs)
          self.check_codegen(f,inputs=inputs)

  def test_mapaccum_schemes(self):

    x = SX.sym("x",2)
    y = SX.sym("y")
    z = SX.sym("z",2,2)
    v = SX.sym("v",Sparsity.upper(3))

    fun = Function("f",[y,z,x,v],[mtimes(z,x)+y,sin(y*x).T,v/y],["y","z","x","v"],["out0","out1","out2"])

    n = 2

    F = fun.mapaccum("map",n,[2],[0])

    scheme_in_fun = fun.name_in()
    scheme_out_fun = fun.name_out()

    scheme_in_F = F.name_in()
    scheme_out_F = F.name_out()

    self.assertTrue(len(scheme_in_fun),len(scheme_in_F))
    self.assertTrue(len(scheme_out_fun),len(scheme_out_F))

    for sf,sF in zip(scheme_in_fun,scheme_in_F):
      self.assertTrue(sf==sF)
    for sf,sF in zip(scheme_out_fun,scheme_out_F):
      self.assertTrue(sf==sF)

    fun = Function("f",[x,y,z,v],[mtimes(z,x)+y,sin(y*x).T,v/y],["x","y","z","v"],["out0","out1","out2"])

    n = 2

    F = fun.mapaccum("map",n)

    self.assertTrue(len(scheme_in_fun),len(scheme_in_F))
    self.assertTrue(len(scheme_out_fun),len(scheme_out_F))

    for sf,sF in zip(scheme_in_fun,scheme_in_F):
      self.assertTrue(sf==sF)
    for sf,sF in zip(scheme_out_fun,scheme_out_F):
      self.assertTrue(sf==sF)

  # @requiresPlugin(Importer,"clang")
  # def test_jitfunction_clang(self):
  #   x = MX.sym("x")
  #   F = Function("f",[x],[x**2],{'jit':True})

  #   out = F([5])
  #   self.checkarray(out[0],25)

  # @requiresPlugin(Importer,"clang")
  # def test_clang_c(self):
  #   compiler = Importer('../data/helloworld.c', 'clang')
  #   f = external("helloworld_c", compiler)
  #   [v] = f([])
  #   self.checkarray(2.37683, v, digits=4)

  # @requiresPlugin(Importer,"clang")
  # def test_clang_cxx(self):
  #   compiler = Importer('../data/helloworld.cxx', 'clang')
  #   f = external("helloworld_cxx", compiler)
  #   [v] = f([])
  #   self.checkarray(2.37683, v, digits=4)

  # @requiresPlugin(Importer,"shell")
  # def test_shell_c(self):
  #   compiler = Importer('../data/helloworld.c', 'shell')
  #   f = external("helloworld_c", compiler)
  #   [v] = f([])
  #   self.checkarray(2.37683, v, digits=4)

  # @requiresPlugin(Importer,"shell")
  # def test_shell_cxx(self):
  #   opts = {'compiler':'g++'}
  #   compiler = Importer('../data/helloworld.cxx', 'shell', opts)
  #   f = external("helloworld_cxx", compiler)
  #   [v] = f([])
  #   self.checkarray(2.37683, v, digits=4)

  def test_depends_on(self):
    x = SX.sym("x")
    y = x**2
    try:
        depends_on(x,y)
    except Exception as e:
        s = str(e)
    self.assertTrue("not symbolic" in s)
    try:
        Function("f",[y],[x])
    except Exception as e:
        s = str(e)
    self.assertTrue("not symbolic" in s)

  def test_1d_interpolant(self):
    grid = [[0, 1, 1.5, 2, 3]]
    values = [0, 1, 2, 5, 3]
    F = interpolant('F', 'linear', grid, values)
    def same(a, b): return abs(float(a)-b)<1e-8
    pairs = [
      (3.4,3-0.4*2),
      (2.4,5-0.4*2),
      (1.6,2+3*0.1/0.5),
      (1.4,1+0.4/0.5),
      (0.4,0.4),
      (-.6,-0.6)
    ]

    X = MX.sym("x")

    J = Function("F",[X],[F(X)])

    for a,r in pairs:
      self.assertTrue(same(F(a), r))
      self.check_codegen(F,inputs=[a],check_serialize=True)
      self.check_serialize(F,[a])
      
    print(F.stats())

    X = MX.sym("x")

    J = Function("F",[X],[jacobian(F(X),X)])

    pairs = [
      (3.4,-2),
      (2.4,-2),
      (1.6,6),
      (1.4,2),
      (0.4,1),
      (-.6,1),

      (1,2),
      (0.99,1),
    ]

    for a,r in pairs:
      self.assertTrue(same(J(a), r))
      self.check_codegen(J,inputs=[a])
      self.check_serialize(J,[a])

  def test_2d_interpolant(self):
    grid = [[0, 1, 4, 5],
            [0, 2, 3]]

    values = [0,   1,  8,  3,
              10, -11, 12, 13,
              20, 31, -42, 53]
    F = interpolant('F', 'linear', grid, values)


    a0 = -11+0.4*(31+11)
    a1 = 12+0.4*(-42-12)
    pairs = [
      (vertcat(1,2), -11),
      (vertcat(1,3), 31),
      (vertcat(4,2), 12),
      (vertcat(4,3), -42),

      (vertcat(1,2.4), a0),
      (vertcat(4,2.4), a1),

      (vertcat(3,2), -11+2.0/3*(12+11)),
      (vertcat(3,3), 31+2.0/3*(-42-31)),

      (vertcat(3,2.4), a0+2.0/3*(a1-a0))
    ]

    for a,r in pairs:
      self.checkarray(F(a), r)
      self.check_codegen(F,inputs=[a])


    X = MX.sym("x",2)

    J = Function("J",[X],[jacobian(F(X),X)])

    jx0 = (12+11)/3.0
    jx1 = (-42-31)/3.0
    jx2 = (13-12)
    jx3 = (53+42)

    jy0 = 31+11
    jy1 = -42-12

    pairs = [
      (vertcat(1,2), vertcat(jx0,jy0)),
      (vertcat(1,3), vertcat(jx1,jy0)),
      (vertcat(4,2), vertcat(jx2,jy1)),
      (vertcat(4,3), vertcat(jx3,jy1)),

      (vertcat(1,2.4), vertcat(jx0+(jx1-jx0)*0.4, 31+11)),
      (vertcat(4,2.4), vertcat(jx2+(jx3-jx2)*0.4, -42-12)),

      (vertcat(3,2), vertcat(jx0,jy0+(jy1-jy0)*2.0/3)),
      (vertcat(3,3), vertcat(jx1,jy0+(jy1-jy0)*2.0/3)),

      (vertcat(3,2.4), vertcat(jx0+(jx1-jx0)*0.4,jy0+(jy1-jy0)*2.0/3)),

    ]

    for a,r in pairs:
      self.checkarray(J(a).T, r)
      self.check_codegen(J,inputs=[a])
      self.check_serialize(J,[a])

  def test_1d_interpolant_uniform(self):
    grid = [[0, 1, 2]]
    values = [0, 1, 2]
    for opts in [{"lookup_mode": ["linear"]},{"lookup_mode": ["exact"]},{"lookup_mode": ["binary"]}]:
      F = interpolant('F', 'linear', grid, values, opts)
      def same(a, b): return abs(float(a)-b)<1e-8
      self.assertTrue(same(F(2.4), 2.4))
      self.assertTrue(same(F(1.4), 1.4))
      self.assertTrue(same(F(0), 0))
      self.assertTrue(same(F(1), 1))
      self.assertTrue(same(F(2), 2))
      self.assertTrue(same(F(6), 6))
      self.assertTrue(same(F(0.4), 0.4))
      self.assertTrue(same(F(-.6), -.6))

      F = interpolant('F', 'linear', [np.linspace(0,1,7)], list(range(7)), {"lookup_mode": ["exact"]})

    grid = [[2, 4, 6]]
    values = [10, 7, 1]
    for opts in [{"lookup_mode": ["linear"]},{"lookup_mode": ["exact"]},{"lookup_mode": ["binary"]}]:
      F = interpolant('F', 'linear', grid, values, opts)
      def same(a, b): return abs(float(a)-b)<1e-8
      self.assertTrue(same(F(1), 11.5))
      self.assertTrue(same(F(2), 10))
      self.assertTrue(same(F(3), 8.5))
      self.assertTrue(same(F(4), 7))
      self.assertTrue(same(F(5), 4))
      self.assertTrue(same(F(6), 1))
      self.assertTrue(same(F(7), -2))

      F = interpolant('F', 'linear', [np.linspace(0,1,7)], list(range(7)), {"lookup_mode": ["exact"]})

  def test_2d_interpolant_uniform(self):
    grid = [[0, 1, 2], [0, 1, 2]]
    values = [0, 1, 2, 10, 11, 12, 20, 21, 22]
    for opts in [{"lookup_mode": ["linear","linear"]},{"lookup_mode": ["exact","exact"]},{"lookup_mode": ["binary","binary"]}]:
      F = interpolant('F', 'linear', grid, values, opts)
      def same(a, b): return abs(float(a)-b)<1e-8
      self.assertTrue(same(F([2.4, 0.5]), 7.4))
      self.assertTrue(same(F([1.4, 0.5]), 6.4))
      self.assertTrue(same(F([0.4, 0.5]), 5.4))
      self.assertTrue(same(F([1, 0.5]), 6))
      self.assertTrue(same(F([1, 0]), 1))
      self.assertTrue(same(F([0, 0]), 0))
      self.assertTrue(same(F([0.4, 1]), 10.4))
      self.assertTrue(same(F([-.6, 0.5]), 4.4))
      self.assertTrue(same(F([-.6, 1.5]), 14.4))
      self.assertTrue(same(F([-.6, 2.5]), 24.4))
      self.assertTrue(same(F([-.6, 3.5]), 34.4))

  @skip(not scipy_interpolate)
  def test_nd_linear(self):

    N = 4
    for n_dim in range(1,5):
      grid = []
      x0 = []
      for i in range(n_dim):
        g = sorted(list(np.random.random(N)))
        grid.append(g)
        x0.append(np.mean(g))

      D = np.random.random([N]*n_dim)

      xref = scipy.interpolate.interpn(grid, D, x0)

      d = D.ravel(order='F')

      F = interpolant('F', 'linear', grid, d)

      self.checkarray(F(x0),xref)


  @skip(not scipy_interpolate)
  def test_2d_bspline(self):
    import scipy.interpolate
    np.random.seed(0)

    d_knots = [list(np.linspace(0,1,5)),list(np.linspace(0,1,6))]

    data = np.random.random([len(e) for e in d_knots])
    r = np.meshgrid(*d_knots,indexing='ij')

    xyz = np.vstack(list(e.ravel(order='F') for e in r)).ravel(order='F')

    d_flat = data.ravel(order='F')

    LUT = casadi.interpolant('name','bspline',d_knots,d_flat)
    LUTJ = jacobian_old(LUT, 0, 0)
    LUTH = hessian_old(LUT, 0, 0)

    self.check_codegen(LUT, [vertcat(0.2,0.3)])
    self.check_serialize(LUT, [vertcat(0.2,0.3)])
    #scipy.interpolate.interpn(d_knots, data, [0.2,0.3], method='splinef2d')

    interp = scipy.interpolate.RectBivariateSpline(d_knots[0], d_knots[1], data)
    for x in [0,0.01,0.1,0.2,0.9,0.99,1]:
      for y in [0,0.01,0.1,0.2,0.9,0.99,1]:
        m = LUT([x,y])
        r = interp.ev(x,y)
        self.checkarray(m,r)

        m = LUTJ([x,y])[0]
        try:
          r = [interp.ev(x,y, 1, 0), interp.ev(x,y, 0, 1)]
        except:
          r = None
        if r is not None:
          self.checkarray(m,r)

        m = LUTH([x,y])[0]
        try:
          r = blockcat([[interp.ev(x,y, 2, 0),interp.ev(x,y, 1, 1)],[interp.ev(x,y, 1, 1), interp.ev(x,y, 0, 2)]])
        except:
          r = None
        if r is not None:
          self.checkarray(m,r)

  def test_1d_bspline_call_fun(self):
  
    fs = []
    for X in [SX,MX]:

        x = X.sym("x")
        np.random.seed(0)

        d_knots = [list(np.linspace(0,1,5))]

        data = np.random.random([len(e) for e in d_knots])
        r = np.array(d_knots)

        xyz = np.vstack(list(e.ravel(order='F') for e in r)).ravel(order='F')

        d_flat = data.ravel(order='F')

        LUT = interpolant('name','bspline',d_knots,d_flat)

        f = Function('f',[x],[2*LUT(sin(x))])
        fs.append(f)
    self.checkfunction(fs[0],fs[1],inputs=[0.32])
        

  @skip(not scipy_interpolate)
  def test_1d_bspline(self):
    import scipy.interpolate
    np.random.seed(0)

    d_knots = [list(np.linspace(0,1,5))]

    data = np.random.random([len(e) for e in d_knots])
    r = np.array(d_knots)

    xyz = np.vstack(list(e.ravel(order='F') for e in r)).ravel(order='F')

    d_flat = data.ravel(order='F')

    LUT = casadi.interpolant('name','bspline',d_knots,d_flat)
    self.check_codegen(LUT, [0.2])
    LUTJ = jacobian_old(LUT, 0, 0)
    LUTH = hessian_old(LUT, 0, 0)

    interp = scipy.interpolate.InterpolatedUnivariateSpline(d_knots[0], data)
    for x in [0,0.01,0.1,0.2,0.9,0.99,1]:
      m = LUT(x)
      r = interp(x)
      self.checkarray(m,r)

      m = LUTJ(x)[0]
      try:
        r = interp(x, 1)
      except:
        r = None
      if r is not None:
        self.checkarray(m,r)

      m = LUTH(x)[0]
      try:
        r = interp(x, 2)
      except:
        r = None
      if r is not None:
        self.checkarray(m,r)

  def test_Callback_Jacobian(self):
    x = MX.sym("x")
    y = MX.sym("y")

    num_inputs = [0.2,0.7]

    g = Function("g", [x,y],[sin(x+3*y)])

    class Fun(Callback):
        # sin(x+3*y)

        def __init__(self):
          Callback.__init__(self)
          self.construct("Fun", {})
        def get_n_in(self): return 2
        def get_n_out(self): return 1

        def eval(self,arg):
          x = arg[0]
          y = arg[1]
          z0 = 3*y
          z1 = x+z0
          z2 = sin(z1)
          return [z2]

        def has_forward(self,nfwd): return False
        def has_reverse(self,nadj): return False

        def has_jacobian(self): return True

        def get_jacobian(self, name, inames, onames, opts):
          x = SX.sym("x")
          y = SX.sym("y")
          out_g = SX.sym('out_g', Sparsity(1,1))
          J = Function(name, [x,y, out_g], [cos(x+3*y), 3*cos(x+3*y)], inames, onames, opts)
          return J

    f = Fun()

    self.checkfunction(f,g,inputs=num_inputs,fwd=False,adj=False,indirect=False)


  def test_Callback_errors(self):

    class Fun(Callback):

        def __init__(self):
          Callback.__init__(self)
          self.construct("Fun", {})
        def get_n_in(self): return 2
        def get_n_out(self): return 1

        def get_sparsity_in(i):
          return 4

        def eval(self,arg):
          x = arg[0]
          y = arg[1]

          z0 = 3*y
          z1 = x+z0
          z2 = sin(z1)
          return [z2]

    try:
      f = Fun()
    except Exception as e:
      s = str(e)
      print(s)
    self.assertTrue("get_sparsity_in" in s)

  def test_Callback(self):

    x = MX.sym("x")
    y = MX.sym("y")

    num_inputs = [0.2,0.7]

    g = Function("g", [x,y],[sin(x+3*y)])

    # Simple syntax
    def getP(indirect=True):

      class Fun(Callback):

        def __init__(self):
          Callback.__init__(self)
          self.construct("Fun", {})

        def get_n_in(self): return 2
        def get_n_out(self): return 1

        def eval(self,arg):
          x = arg[0]
          y = arg[1]

          z0 = 3*y
          z1 = x+z0
          z2 = sin(z1)
          return [z2]

      self.cb = Fun()

      if not indirect:
        return self.cb

      f = Function("f", [x,y],[self.cb(x,y)])

      return f

    for indirect in [True,False]:
      f = getP(indirect=indirect)
      self.checkfunction(f,g,inputs=num_inputs,sens_der=False,jacobian=False,gradient=False,hessian=False,evals=False)

      with self.assertRaises(Exception):
        f.gradient()

      with self.assertRaises(Exception):
        jacobian_old(f, 0, 0)

      with self.assertRaises(Exception):
        f.forward(1)
      with self.assertRaises(Exception):
        f.reverse(1)

  def test_Callback_dimcheck(self):
      class Fun(Callback):
        def __init__(self):
          Callback.__init__(self)
          self.construct("Fun")
        def get_n_in(self): return 2
        def get_n_out(self): return 1

        def eval(self,arg):
          return [2, 1]
      f = Fun()

      s = ""
      try:
        f(2)
      except Exception as e:
        s = str(e)
      self.assertTrue("Incorrect number of inputs" in s)
      class Fun(Callback):
        def __init__(self):
          Callback.__init__(self)
          self.construct("Fun")
        def get_n_in(self): return 2
        def get_n_out(self): return 1

        def eval(self,arg):
          return [2, 1]
      f = Fun()

      s = ""
      try:
        f(2,3)
      except Exception as e:
        s = str(e)
      self.assertTrue("Expected 1 output" in s)
      s = ""
      class Fun(Callback):
        def __init__(self):
          Callback.__init__(self)
          self.construct("Fun")
        def get_n_in(self): return 2
        def get_n_out(self): return 1

        def eval(self,arg):
          return [DM.zeros(2,2)]
      f = Fun()
      try:
        f(2,3)
      except Exception as e:
        s = str(e)
      self.assertTrue("Shape mismatch" in s)

  def test_Callback_sens(self):
    x = MX.sym("x")
    y = MX.sym("y")

    num_inputs = [0.2,0.7]

    g = Function("g", [x,y],[sin(x+3*y)])

    def getP(has_fwd=True,has_adj=True,indirect=True):

      class Fun(Callback):
        # sin(x+3*y)

        def __init__(self,opts):
          Callback.__init__(self)
          self.construct("Fun", opts)
        def get_n_in(self): return 2
        def get_n_out(self): return 1

        def eval(self,arg):
          x = arg[0]
          y = arg[1]
          z0 = 3*y
          z1 = x+z0
          z2 = sin(z1)
          return [z2]

        if has_fwd:
          def has_forward(self,nfwd): return nfwd==1
          def get_forward(self,nfwd,name,inames,onames,opts):
            assert(nfwd==1)
            class ForwardFun(Callback):
              # sin(x+3*y)

              def __init__(self):
                Callback.__init__(self)
                self.construct(name, {"verbose":True})
              def get_n_in(self): return 2+1+2
              def get_n_out(self): return 1

              def eval(self,arg):
                x,y = arg[0],arg[1]
                z = arg[2]
                seeds = arg[3:]

                z0 = 3*y
                z1 = x+z0
                z2 = sin(z1)

                ret = []

                for i in range(3,len(arg),2):
                  dx = arg[i]
                  dy = arg[i+1]
                  dz0 = 3*dy
                  dz1 = dx+dz0
                  dz2 = cos(z1)*dz1
                  ret.append(dz2)

                return ret

            self.cb_fwd = ForwardFun()
            return self.cb_fwd

        if has_adj:
          def has_reverse(self,nadj): return nadj==1
          def get_reverse(self,nadj,name,inames,onames,opts):
            assert(nadj==1)
            class BackwardFun(Callback):
              # sin(x+3*y)

              def __init__(self):
                Callback.__init__(self)
                self.construct(name, {"verbose":True})
              def get_n_in(self): return 2+1+1
              def get_n_out(self): return 2

              def eval(self,arg):
                x,y = arg[0],arg[1]
                z = arg[2]
                seeds = arg[3:]

                z0 = 3*y
                z1 = x+z0
                z2 = sin(z1)

                ret = []

                for i in range(3,len(arg)):
                  z_bar = arg[i]
                  bx = 0
                  by = 0
                  bz1 = 0
                  bz0 = 0

                  bz2 = z_bar
                  bz1 += bz2*cos(z1)
                  bx+= bz1;bz0+= bz1
                  by+= 3*bz0
                  ret.append(bx)
                  ret.append(by)
                return ret

            self.cb_rev = BackwardFun()
            return self.cb_rev

      opts = {"verbose":True}
      self.cb = Fun(opts)
      f = self.cb

      if not indirect:
        return f

      f = Function("f", [x,y],[f(x,y)])

      return f

    for indirect in [True,False]:
      f = getP(has_fwd=True,has_adj=True,indirect=indirect)

      self.checkfunction(f,g,inputs=num_inputs,sens_der=False,hessian=False,evals=1)

      f = getP(has_fwd=True,has_adj=False,indirect=indirect)

      self.checkfunction(f,g,inputs=num_inputs,sens_der=False,hessian=False,adj=False,evals=1)

      f = getP(has_fwd=False,has_adj=True,indirect=indirect)

      self.checkfunction(f,g,inputs=num_inputs,sens_der=False,hessian=False,fwd=False,evals=1)

      f = getP(has_fwd=True,has_adj=False,indirect=indirect)

      self.checkfunction(f,g,inputs=num_inputs,sens_der=False,hessian=False,adj=False,evals=1)

      f = getP(has_fwd=False,has_adj=True,indirect=indirect)

      self.checkfunction(f,g,inputs=num_inputs,sens_der=False,hessian=False,fwd=False,evals=1)

  def test_callback_jacobian_sparsity(self):

    x = MX.sym("x",2)

    h = 1e-7
    for with_jac_sparsity in [True, False]:

      calls = []

      class Fun(Callback):

          def __init__(self):
            Callback.__init__(self)
            self.construct("Fun", {"enable_fd":True,"fd_method":"forward","fd_options":{"h": h,"h_iter":False}})
          def get_n_in(self): return 1
          def get_n_out(self): return 1
          def get_sparsity_in(self,i): return Sparsity.dense(2,1)
          def get_sparsity_out(self,i): return Sparsity.dense(2,1)

          def eval(self,arg):
            x = arg[0]
            calls.append((x-DM([5,7]))/h)
            return [x**2]

          def has_forward(self,nfwd): return False
          def has_reverse(self,nadj): return False

          def has_jac_sparsity(self,oind,iind): return with_jac_sparsity

          def get_jac_sparsity(self,oind,iind,symmetric):
            assert(oind == 0)
            assert(iind == 0)
            return Sparsity.diag(2)

      f = Fun()
      y = jacobian(f(x),x)

      F = Function('F',[x],[y])

      with capture_stdout() as out:
       J = F([5,7])
      calls = hcat(calls)
      J_ref = DM([[5*2,0],[0,7*2]])
      if with_jac_sparsity:
        self.checkarray(calls,DM([[0,0],[1,1]]).T,digits=5)
        J_ref = sparsify(J_ref)
      else:
        self.checkarray(calls,DM([[0,0],[1,0],[0,1]]).T,digits=5)

      self.checkarray(J,J_ref,digits=5)
      self.assertTrue(J.sparsity()==J_ref.sparsity())


  @requires_nlpsol("ipopt")
  def test_common_specific_options(self):

      x = SX.sym("x")

      nlp = {"x": x, "f": x**4}

      with capture_stdout() as out:
        solver = nlpsol("solver","ipopt",nlp)
      self.assertTrue("nlp_f" not in out[0])
      solver = nlpsol("solver","ipopt",nlp,{"common_options":{"final_options" : {"print_in":True}}})
      with capture_stdout() as out:
        solver(x0=0.1)
      print(out[0])
      self.assertTrue("Function nlp_f" in out[0])
      self.assertTrue("Function nlp_grad_f" in out[0])
      solver = nlpsol("solver","ipopt",nlp,{"specific_options":{ "nlp_f" : {"final_options" : {"print_in":True}}}})
      with capture_stdout() as out:
        solver(x0=0.1)
      self.assertTrue("Function nlp_f" in out[0])
      self.assertFalse("Function nlp_grad_f" in out[0])
      solver = nlpsol("solver","ipopt",nlp,{"common_options":{"final_options" : {"print_in":True}},"specific_options":{ "nlp_f" : {"final_options" : {"print_in":False}}}})
      with capture_stdout() as out:
        solver(x0=0.1)
      self.assertFalse("Function nlp_f" in out[0])
      self.assertTrue("Function nlp_grad_f" in out[0])
      solver = nlpsol("solver","ipopt",nlp,{"common_options":{"final_options" : {"print_in":False}},"specific_options":{ "nlp_f" : {"final_options" : {"print_in":True}}}})
      with capture_stdout() as out:
        solver(x0=0.1)
      self.assertTrue("Function nlp_f" in out[0])
      self.assertFalse("Function nlp_grad_f" in out[0])

      with capture_stdout() as out:
        solver = nlpsol("solver","ipopt",nlp)
      self.assertTrue(len(out[1])==0)
      with capture_stdout() as out:
        solver = nlpsol("solver","ipopt",nlp,{"specific_options":{ "nlp_foo" : {"helper_options" : {"verbose":True}}}})
      self.assertTrue("Ignoring" + out[1])
      self.assertTrue("nlp_g" in out[1])
      with self.assertRaises(Exception):
        solver = nlpsol("solver","ipopt",nlp,{"specific_options":{ "nlp_foo" : 3}})


  @requires_nlpsol("ipopt")
  def test_oracle_options(self):
    DM.set_precision(16)

    N = 5
    x = MX.sym("x",N,1)

    options = {}
    options["ipopt"] = {"hessian_approximation":"limited-memory"}
    options["oracle_options"] = {"enable_fd":True,"enable_forward":False,"enable_reverse":False,"print_in":True,"fd_method":"central"}

    solver = nlpsol("solver","ipopt",{"x":x,"f":dot(x,x)},options)

    with capture_stdout() as out:
      solver(x0=1)


    DM.set_precision(6)

    self.assertTrue("[1, 1, 1, 1, 1]" in out[0])


  @requires_expm("slicot")
  @memory_heavy()
  def test_expm(self):
      eps = 1e-6
      t = MX.sym('t')
      tnum = 0.2

      n = 4

      np.random.seed(0)
      Anum = np.random.random((n,n))
      Bnum = np.random.random((n,2))
      Bb = np.random.random((n,2))

      dA = np.random.random((n,n))
      Yb = np.random.random((n,2))

      def expm(A):

        n = A.shape[0]
        x = MX.sym('x',n)
        As = MX.sym('A',n,n)

        dae = {'x':x,'p':vec(As),'ode':mtimes(As,x)}
        intg = integrator('intg','cvodes',dae,{'reltol':1e-14,'abstol':1e-14})

        Intg = intg.map('identity','serial',n,[1],[])

        out = Intg(x0=DM.eye(n),p=vec(As))
        expmF = Function('expm',[As],[out["xf"]])
        return expmF(A)

      A = MX.sym("A",n,n)
      t = MX.sym("t")
      fr = Function('fr',[A,t],[expm(A*t)])
      f = Function('f',[A,t],[casadi.expm(A*t)])

      self.checkfunction(fr,f,inputs=[Anum, 1.1],digits=8)

      fr = Function('fr',[t],[expm(Anum*t)])
      f = Function('f',[t],[casadi.expm_const(Anum,t)])

      self.checkfunction(fr,f,inputs=[1.1],digits=8)

      JA = jacobian(casadi.expm(A*t),A)
      Jt = jacobian(casadi.expm(A*t),t)

      self.assertTrue(JA.nnz()==n**4)
      self.assertTrue(Jt.nnz()==n**2)

      JA = jacobian(casadi.expm_const(A,t),A)
      Jt = jacobian(casadi.expm_const(A,t),t)

      self.assertTrue(JA.nnz()==0)
      self.assertTrue(Jt.nnz()==n**2)

  def test_conditional(self):

    np.random.seed(5)

    x = MX.sym('x',2,2)
    y = MX.sym('y',2,2)

    sp1 = MX.sym('y',Sparsity.lower(2))
    sp2 = MX.sym('z',Sparsity.diag(2))

    f1 = Function("f",[sp2,x],[x**2,x*sp2])
    f2 = Function("f",[sp1,x],[2*x**2,sin(sp1)])
    f3 = Function("f",[sp1,sp2],[sp1*sp2,sp1+sp2])

    F = Function.conditional("test",[f1,f2], f3)
    Fsx = F.expand()

    A = np.random.random((2,2))
    B = np.random.random((2,2))

    for i in range(-1,3):
      self.checkfunction(F,Fsx,inputs = [i,A,B])
      self.check_codegen(F,inputs=[i,A,B])

  def test_max_num_dir(self):
    x = MX.sym("x",10)

    f = Function("ffff",[x],[mtimes(DM.ones(10,10),x)],{"max_num_dir":4,"verbose":True})
    f = f.expand()


    y = f.call([sin(x)],False,True)[0]



    acc = Function('acc',[x],[y])
    acc = acc.mapaccum('accd',5)


    cons = vec(acc(x))

    os.system("callgrind_control -i on")

    e = jacobian(cons,x)
    os.system("callgrind_control -i off")

    g = Function('g',[x],[e])


    c = CodeGenerator('me')
    c.add(g)
    code= c.dump()

    self.assertTrue("fwd4_ffff" in code)

  def test_max_num_dir(self):
    x = MX.sym("x")

    f = Function("ffff",[x],[sin(x)])
    fm = f.mapaccum("mapaccum",100,1,{"base":4})

    c = CodeGenerator('me')
    c.add(fm)
    code= c.dump()

    self.assertTrue("ffff_acc4_acc4_acc4" in code)
    
    
  def test_codegen_with_jac_sparsity(self):
  
    if not args.run_slow: return
    x = MX.sym("x",3)
    y = MX.sym("y",3,3)
    f = Function("f",[x,y],[x**3,mtimes(y,x)])
    c = CodeGenerator('me')
    c.add(f, True)
    
    fJ = f.jacobian()
    
    DM.rng(0)
    x0 = DM.rand(x.sparsity())
    y0 = DM.rand(y.sparsity())
    [F,_] = self.check_codegen(f,inputs=[x0,y0],with_jac_sparsity=True,external_opts={"enable_fd":True})
    FJ = F.jacobian()
    for i in range(fJ.n_out()):
        self.assertEqual(fJ.sparsity_out(i),FJ.sparsity_out(i))
    [F,_] = self.check_codegen(f,inputs=[x0,y0],with_jac_sparsity=False,external_opts={"enable_fd":True})
    FJ = F.jacobian()
    for i in range(fJ.n_out()):
        self.assertTrue(FJ.sparsity_out(i).is_dense())

  def test_2d_linear_multiout(self):
    np.random.seed(0)

    d_knots = [list(np.linspace(0,1,5)),list(np.linspace(0,1,6))]

    data0 = np.random.random([len(e) for e in d_knots])
    data1 = np.random.random([len(e) for e in d_knots])
    r = np.meshgrid(*d_knots,indexing='ij')

    xyz = np.vstack(list(e.ravel(order='F') for e in r)).ravel(order='F')

    d_flat0 = data0.ravel(order='F')

    LUT0 = casadi.interpolant('name','linear',d_knots,d_flat0)

    d_flat1 = data1.ravel(order='F')

    LUT1 = casadi.interpolant('name','linear',d_knots,d_flat1)

    data = np.vstack((data0.ravel(order='F'),data1.ravel(order='F'))).ravel(order='F')

    d_flat = data.ravel(order='F')


    LUT = casadi.interpolant('name','linear',d_knots,d_flat)


    x = MX.sym("x")
    y = MX.sym("y")


    LUT_sep = Function('f',[x,y],[vertcat(LUT0(vertcat(x,y)),LUT1(vertcat(x,y)))])
    LUT = Function('f',[x,y],[LUT(vertcat(x,y))])

    self.checkfunction(LUT,LUT_sep, inputs=[0.2,0.333])
    self.check_codegen(LUT,inputs=[0.2,0.333])
    self.check_serialize(LUT,inputs=[0.2,0.333])

    LUT_param = casadi.interpolant('name','linear',d_knots,2)
    f = Function('LUTp',[x,y],[LUT_param(vertcat(x,y),d_flat)])
    self.checkfunction(LUT,f, inputs=[0.2,0.333])
    self.check_codegen(f,inputs=[0.2,0.333])
    self.check_serialize(f,inputs=[0.2,0.333])

    d_knots_cat = vcat(d_knots[0]+d_knots[1])

    LUT_param = casadi.interpolant('name','linear',[5,6],2)
    f = Function('LUTp',[x,y],[LUT_param(vertcat(x,y),d_knots_cat,d_flat)])
    self.checkfunction(LUT,f, inputs=[0.2,0.333])
    self.check_codegen(f,inputs=[0.2,0.333])
    self.check_serialize(f,inputs=[0.2,0.333])

    LUT_param = casadi.interpolant('name','linear',[5,6],d_flat)
    f = Function('LUTp',[x,y],[LUT_param(vertcat(x,y),d_knots_cat)])
    self.checkfunction(LUT,f, inputs=[0.2,0.333])
    self.check_codegen(f,inputs=[0.2,0.333])
    self.check_serialize(f,inputs=[0.2,0.333])

  def test_2d_bspline_multiout(self):
    np.random.seed(0)

    d_knots = [list(np.linspace(0,1,5)),list(np.linspace(0,1,6))]

    data0 = np.random.random([len(e) for e in d_knots])
    data1 = np.random.random([len(e) for e in d_knots])
    r = np.meshgrid(*d_knots,indexing='ij')

    xyz = np.vstack(list(e.ravel(order='F') for e in r)).ravel(order='F')

    d_flat0 = data0.ravel(order='F')

    LUT0 = casadi.interpolant('name','bspline',d_knots,d_flat0)

    d_flat1 = data1.ravel(order='F')

    LUT1 = casadi.interpolant('name','bspline',d_knots,d_flat1)

    data = np.vstack((data0.ravel(order='F'),data1.ravel(order='F'))).ravel(order='F')

    d_flat = data.ravel(order='F')


    LUT = casadi.interpolant('name','bspline',d_knots,d_flat)


    x = MX.sym("x")
    y = MX.sym("y")


    LUT_sep = Function('f',[x,y],[vertcat(LUT0(vertcat(x,y)),LUT1(vertcat(x,y)))])
    LUT = Function('f',[x,y],[LUT(vertcat(x,y))])

    self.checkfunction(LUT,LUT_sep, inputs=[0.2,0.333])
    self.check_codegen(LUT,inputs=[0.2,0.333])
    self.check_serialize(LUT,inputs=[0.2,0.333])

    LUT_param = casadi.interpolant('name','bspline',d_knots,2)
    f = Function('LUTp',[x,y],[LUT_param(vertcat(x,y),d_flat)])
    self.checkfunction(LUT,f, inputs=[0.2,0.333])
    self.check_codegen(f,inputs=[0.2,0.333])
    self.check_serialize(f,inputs=[0.2,0.333])


  def test_parametric_bspline(self):
    knots = [[0,0,0,0,0.2,0.5,0.8,1,1,1,1],[0,0,0,0.1,0.5,0.9,1,1,1]]
    x=MX.sym("x",2)
    data = np.random.random((7,6,2)).ravel(order='F')
    y = bspline(x,data,knots,[3,2],2)

    C = MX.sym("C",data.shape[0],1)
    Y = bspline(x,C,knots,[3,2],2)
    f = Function('f',[x],[y])
    F = Function('f',[x,C],[Y])
    F = Function('f',[x],[F(x,data)])
    self.checkfunction(f,F,inputs=[vertcat(0.3,0.4)])
    self.check_codegen(F,inputs=[vertcat(0.3,0.4)])
    self.check_serialize(F,inputs=[vertcat(0.3,0.4)])


  def test_smooth_linear(self):
    np.random.seed(0)

    d_knots = [list(np.linspace(0,1,5)),list(np.linspace(0,1,6))]

    data0 = np.random.random([len(e) for e in d_knots])
    data1 = np.random.random([len(e) for e in d_knots])
    r = np.meshgrid(*d_knots,indexing='ij')

    data = np.vstack((data0.ravel(order='F'),data1.ravel(order='F'))).ravel(order='F')

    d_flat = data.ravel(order='F')


    LUT_linear = casadi.interpolant('name','linear',d_knots,d_flat)


    LUT = casadi.interpolant('name','bspline',d_knots,d_flat,{"algorithm": "smooth_linear","smooth_linear_frac":0.1})
    LUT2 = casadi.interpolant('name','bspline',d_knots,d_flat,{"algorithm": "smooth_linear","smooth_linear_frac":0.05})


    # Far away from points: almost identical
    diff1 = float(norm_1(LUT([0.2,0.333])-LUT_linear([0.2,0.333])))
    # Large difference near the knots
    diff2 = float(norm_2(LUT([0.25,0.4])-LUT_linear([0.25,0.4])))
    diff22 = float(norm_2(LUT2([0.25,0.4])-LUT_linear([0.25,0.4])))

    self.assertTrue(diff1<=1e-10)
    self.assertTrue(diff2>=100*diff1)
    self.assertTrue(diff22<=diff2*0.51)
    self.assertTrue(diff22>=diff2*0.49)

    self.checkarray(LUT2([0.26,0.39]),DM([0.9261362392504342, 0.9157189108791507]))


    xy = MX.sym("xy",2)
    LUT_param = casadi.interpolant('name','bspline',d_knots,2,{"algorithm": "smooth_linear","smooth_linear_frac":0.1})
    f = Function('LUTp',[xy],[LUT_param(xy,d_flat)])
    self.checkfunction(LUT,f, inputs=[vertcat(0.2,0.333)])
    self.check_codegen(f,inputs=[vertcat(0.2,0.333)],main=True)
    self.check_serialize(f,inputs=[vertcat(0.2,0.333)])


  def test_codegen_avoid_stack(self):
    x = SX.sym("x",3,3)
    f = Function('f',[x],[det(x)])
    np.random.seed(0)
    self.check_codegen(f,inputs=[np.random.random((3,3))])
    self.check_codegen(f,inputs=[np.random.random((3,3))], opts={"avoid_stack": True})


  def test_serialize(self):
    for opts in [{"debug":True},{}]:
      x = SX.sym("x")
      y = x+3
      z = sin(y)

      f = Function('f',[x],[z])
      fs = Function.deserialize(f.serialize(opts))

      self.checkfunction(f,fs,inputs=[2])

      x = SX.sym("x")
      y = x+3
      z = sin(y)

      f = Function('f',[x],[z,np.nan,-np.inf,np.inf])
      fs = Function.deserialize(f.serialize(opts))
      self.checkfunction(f,fs,inputs=[2])

      x = SX.sym("x")
      y = SX.sym("y", Sparsity.lower(3))
      z = x+y
      z1 = sparsify(vertcat(z[0],0,z[1]))
      z2 = z.T

      f = Function('f',[x,y],[z1,z2,x**2],["x","y"],["a","b","c"])
      fs = Function.deserialize(f.serialize(opts))

      self.assertEqual(fs.name_in(0), "x")
      self.assertEqual(fs.name_out(0), "a")
      self.assertEqual(fs.name(), "f")

      self.checkfunction(f,fs,inputs=[3.7,np.array([[1,0,0],[2,3,0],[4,5,6]])],hessian=False)


      fs = Function.deserialize(f.serialize(opts))
      self.checkfunction(f,fs,inputs=[3.7,np.array([[1,0,0],[2,3,0],[4,5,6]])],hessian=False)

      x = SX.sym("x")
      p = SX.sym("p")

      f = Function('f',[x],[p], {"allow_free":True})

      #SXFunction with free parameters
      pickle.loads(pickle.dumps(f))


      x = MX.sym("x")
      f = Function('f',[x],[x**2])

      fs = Function.deserialize(f.serialize(opts))
      self.checkfunction(f,fs,inputs=[3.7],hessian=False)


      x = MX.sym("x")
      y = MX.sym("y",2)

      w = if_else(x, atan2(3*norm_fro(y)*y,x), x-y, True)
      z = sin(2*x)*w[0]+1
      g = Function("g",[x,y],[w-x])
      gmap = g.map(2, "thread", 2)
      gmapsx = gmap.expand()

      q = gmap(horzcat(2*x,x-y[1]),horzcat(z+y,cos(z+y)))+1/gmapsx(horzcat(2*x,x-y[1]),repmat(z+y,1,2))

      q = solve(q,2*y, "qr")
      q+= bilin(DM([[1,3],[7,8]]),q,2*q)

      f = Function("f",[x,y],[q+1,jacobian(q, vertcat(x, y))])

      fs = Function.deserialize(f.serialize(opts))
      self.checkfunction(f,fs,inputs=[1.1, vertcat(2.7,3)],hessian=False)

  @memory_heavy()
  def test_serialize_recursion_limit(self):
      for X in [SX,MX]:
        x = X.sym("x")

        y = x
        for i in range(10000):
          y = sin(y)

        f = Function('foo',[x],[y])
        Function.deserialize(f.serialize())


  def test_string(self):
    x=MX.sym("x")
    y=MX.sym("y")
    f = Function('f',[x],[],["x"],[])
    self.assertTrue("(x)->()" in str(f))

    f = Function('f',[],[x],[],["y"], {"allow_free": True})
    self.assertTrue("()->(y)" in str(f))

    f = Function('f',[x],[x**2],["x"],["y"])
    self.assertTrue("(x)->(y)" in str(f))

    f = Function('f',[x,y],[x**2,x*y],["x","y"],["w","z"])
    self.assertTrue("(x,y)->(w,z)" in str(f))

  def test_fold(self):
    x = SX.sym("x",2,2)

    f = Function("f",[x],[sin(x)])

    F = f.fold(10)

    x0 = x
    for i in range(10):
      x = f(x)
    Fref = Function("f",[x0],[x])

    self.checkfunction(F,Fref,inputs=[DM([[1,2],[3,7]])])


  @memory_heavy()
  def test_thread_safety(self):
    x = MX.sym('x')
    y = MX.sym('y')
    f = Function('f', [x, y], [x ** 2 - y])
    for rf in ["newton","fast_newton"]:
      finv = rootfinder('finv', rf, f)

      finv_par = finv.map(50,"unroll").map(4, 'thread',4)
      res = finv_par(numpy.ones(200), numpy.linspace(0, 10, 200))
      self.checkarray(norm_inf(res.T-sqrt(numpy.linspace(0, 10, 200))),0, digits=5)

      finv_par = finv.map(200, 'thread',4)
      res = finv_par(numpy.ones(200), numpy.linspace(0, 10, 200))
      self.checkarray(norm_inf(res.T-sqrt(numpy.linspace(0, 10, 200))),0, digits=5)

  def test_mapped_eval(self):
      x = SX.sym('x')
      y = SX.sym('y', 2)
      f = Function('f', [x,y], [sin(x)*y], ['x','y'], ['r'])
      r1 = f(1, DM([3,4]))
      r2 = f(2, DM([3,4]))
      r3 = f(3, DM([3,4]))
      r_all = f(DM([[1,2,3]]), DM([3,4]))
      self.checkarray(r_all, horzcat(r1,r2,r3), "Mapped evaluation (DM)")

      z = MX.sym("z", 1, 3);
      rz = f(z, DM([3,4]))
      F = Function('F', [z], [rz]);
      r_mx = F(DM([[1,2,3]]))
      self.checkarray(r_all, r_mx, "Mapped evaluation (MX)")

  def test_default_arg(self):
      x = MX.sym("x")
      y = MX.sym("y")
      z = MX.sym("z")

      f = Function('f',[x,y,z],[x,y,z],["x","y","z"],["a","b","c"],{"default_in": [1,2,3]})
      self.check_codegen(f,{"x":5,"z":3},main=True)

  def test_factory_inherit_options(self):
      x = MX.sym("x",5)


      for op in ["grad:f:x","jac:f:x","triu:hess:f:x:x"]:
        f = Function("f",[x],[dot(x,x)],["x"],["f"],{"verbose": True})

        with capture_stdout() as out:
          fgrad = f.factory("fgrad",["x"],[op])

        self.assertTrue("::init" in out[0])


        f = Function("f",[x],[dot(x,x)],["x"],["f"],{"enable_fd":True,"enable_forward":False,"enable_reverse":False,"print_in":True,"fd_method":"central","fd_options":{"h": 1e-7,"h_iter":False}})

        fgrad = f.factory("fgrad",["x"],[op])

        with capture_stdout() as out:
          fgrad(0)

        self.assertTrue("-1e-07," in out[0] or "-1e-007," in out[0] )
        self.assertTrue("1e-07," in out[0] or "1e-007," in out[0] )

  @requires_nlpsol("ipopt")
  @requiresPlugin(Importer,"shell")
  def test_inherit_jit_options(self):

    x = MX.sym("x")

    f = Function('Q',[x],[x**2],{"jit":True,"compiler":"shell","verbose":True})

    [g] = f.call([sin(x)],False,True)
    g = cos(g)
    y = MX.sym("x")
    with capture_stdout() as out:
      solver = nlpsol("solver","ipopt",{"x": x,"f": x**2, "g": g})

    import re
    found = set([m.group(1) for m in re.finditer("Compiling function '(\w+?)'", out[0])])
    self.assertTrue(len(found)==1)
    self.assertTrue("fwd1_Q" in found)

    with capture_stdout() as out:
      self.check_serialize(solver,inputs={"x0": 2})

    found = set([m.group(1) for m in re.finditer("Compiling function '(\w+?)'", out[0])])
    self.assertTrue(len(found)==2)
    self.assertTrue("Q" in found)
    self.assertTrue("fwd1_Q" in found)

  def test_custom_jacobian(self):
    x = MX.sym("x")
    p = MX.sym("p")
    J = Function("jac_Q", [x, p, MX(1,1), MX(1,1)], [x*pi, 1, 1, 1],
        ['x', 'p', 'out_r', 'out_s'], ['jac_r_x', 'jac_r_p', 'jac_s_x', 'jac_s_p'])

    f = Function('Q', [x, p], [x**2, 2*x*p], ['x', 'p'], ['r', 's'],
            dict(custom_jacobian = J, jac_penalty = 0))

    J = None

    [g1,g2] = f.call([x,p],False,True)
    JJ = Function('JJ',[x,p],[jacobian(g1+3*x,x)])

    self.checkarray(JJ(5,2),5*pi+3)

  def test_dump(self):
    x = MX.sym("x",Sparsity.lower(3))
    y = MX.sym("y",0,0)
    z = MX.sym("z",2,2)

    for fmt in ["txt","mtx"]:
      f = Function("f",[x,y,z],[2*x,2*z],["x","y","z"],["a","c"],{"dump":True,"dump_in":True,"dump_out":True,"dump_format":fmt})
      ins = [sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM(),DM([[1,3],[4,5]])]
      out = f(*ins)

      F = Function.load("f.casadi")

      x_num = DM.from_file("f.000000.in.x." + fmt)
      y_num = DM.from_file("f.000000.in.y." + fmt)
      z_num = DM.from_file("f.000000.in.z." + fmt)

      self.checkarray(x_num,ins[0])
      self.checkarray(y_num,ins[1])
      self.checkarray(z_num,ins[2])

      a_num = DM.from_file("f.000000.out.a." + fmt)
      c_num = DM.from_file("f.000000.out.c." + fmt)

      self.checkarray(a_num,out[0])
      self.checkarray(c_num,out[1])


      F.generate_in("test_in.txt", [x_num,y_num,z_num])
      F.generate_out("test_out.txt", F.call([x_num,y_num,z_num]))
      X = DM.from_file("f.000000.in.txt")
      A = DM.from_file("f.000000.out.txt")

      Xr = DM.from_file("test_in.txt")
      Ar = DM.from_file("test_out.txt")

      self.checkarray(Xr,X)
      self.checkarray(Ar,A)

  def test_eval_shapes(self):
    x = MX.sym("x",Sparsity.lower(3))
    y = MX.sym("y",3,1)
    z = MX.sym("z",2,2)
    f = Function("f",[x,y,z],[2*x,2*y,2*z],{"default_in":[1,2,3]})

    for ins, ref, ref_flat in [
      #normal
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM([9,8,7]),DM([[1,3],[4,5]])],
       [2*sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])), 2*DM([9,8,7]), 2*DM([[1,3],[4,5]])],
       [1,2,7,4,8,9,9,8,7,1,4,3,5]),

      #Project
      ([sparsify(DM([[1,0,0],[2,0,0],[7,8,9]])),DM([9,8,7]),DM([[1,3],[4,5]])],
       [2*sparsify(DM([[1,0,0],[2,1e-30,0],[7,8,9]])),2*DM([9,8,7]),2*DM([[1,3],[4,5]])],
       [1,2,7,0,8,9,9,8,7,1,4,3,5]),

      #Project away
      ([sparsify(DM([[1,0,1],[2,4,0],[7,8,9]])),DM([9,8,7]),DM([[1,3],[4,5]])],
       [2*sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])), 2*DM([9,8,7]), 2*DM([[1,3],[4,5]])],
       [1,2,7,4,8,9,9,8,7,1,4,3,5]),

      #Scalar expansion
      ([DM(4),5,7],
       [2*sparsify(DM([[4,0,0],[4,4,0],[4,4,4]])), 2*DM([5,5,5]), 2*DM([[7,7],[7,7]])],
       [4,4,4,4,4,4,5,5,5,7,7,7,7]),


      #Repmat
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM([9,8,7]),DM([1,3])],
       [2*sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])), 2*DM([9,8,7]), 2*DM([[1,1],[3,3]])],
       [1,2,7,4,8,9,9,8,7,1,3,1,3]),

      #npar
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM([[9,2],[8,4],[7,4]]),DM([[1,3],[4,5]])],
       [2*sparsify(DM([[1,0,0,1,0,0],[2,4,0,2,4,0],[7,8,9,7,8,9]])), 2*DM([[9,2],[8,4],[7,4]]), 2*DM([[1,3,1,3],[4,5,4,5]])],
       None),


      # growing npar
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM([[9,2],[8,4],[7,4]]),DM([[1,3,4,5,1,1,1,1],[4,5,7,8,3,3,3,3]])],
       [2*repmat(sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),1,4), 2*repmat(DM([[9,2],[8,4],[7,4]]),1,2), 2*DM([[1,3,4,5,1,1,1,1],[4,5,7,8,3,3,3,3]])],
       None),

      # growing npar
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM([[9,2,3,4],[8,4,7,8],[7,4,9,10]]),DM([[1,3,4,5],[4,5,7,8]])],
       None,
       None),

      #Transpose
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM([9,8,7]).T,DM([[1,3],[4,5]])],
       [2*sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),2*DM([9,8,7]), 2*DM([[1,3],[4,5]])],
       [1,2,7,4,8,9,9,8,7,1,4,3,5]),

      #Null: note default_in applies only to dict style
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM(),DM([1,3])],
       [2*sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),2*DM([0,0,0]), 2*DM([[1,1],[3,3]])],
       [1,2,7,4,8,9,0,0,0,1,3,1,3]),

      #Null: note default_in applies only to dict style
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM(3,0),DM([1,3])],
       [2*sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),2*DM([0,0,0]), 2*DM([[1,1],[3,3]])],
       [1,2,7,4,8,9,0,0,0,1,3,1,3]),

      #Null: note default_in applies only to dict style
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM(0,3),DM([1,3])],
       [2*sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),2*DM([0,0,0]), 2*DM([[1,1],[3,3]])],
       [1,2,7,4,8,9,0,0,0,1,3,1,3]),

      #Null: should be forbidden
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM(0,5),DM([1,3])],
       [2*sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),2*DM([0,0,0]), 2*DM([[1,1],[3,3]])],
       [1,2,7,4,8,9,0,0,0,1,3,1,3]),

      #Null: should be forbidden
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM(8,0),DM([1,3])],
       [2*sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),2*DM([0,0,0]), 2*DM([[1,1],[3,3]])],
       [1,2,7,4,8,9,0,0,0,1,3,1,3]),



      # Wrong
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM([9,8,7]).T,DM([1,3]).T],None,None),
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9],[4,5,6]])),DM([9,8,7]),DM([1,3])],None,None),
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM([[9,2,1],[8,4,1],[7,4,1]]),DM([[1,1,1,3],[3,4,5,6]])],None,None),
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM([9,8,7]),DM([[1,1,1],[3,4,5]])],None,None),
      ([sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])),DM([9,8,7,6]),DM([[1,3],[4,5]])],None,None),

      ]:

      if ref is None:
        with self.assertInException("mismatching shape"):
          res = f.call(ins)
      else:
        res = f.call(ins)

        for i in range(f.n_in()):
          self.assertTrue(res[i].sparsity()==ref[i].sparsity())
          self.checkarray(res[i],ref[i])

        if ref_flat is None:
          with self.assertInException("mismatching shape"):
            f.nz_from_out(res)
        else:
          res_flat = f.nz_from_out(res)

          self.checkarray(res_flat,[i*2 for i in ref_flat])

          res_re = f.nz_to_out(res_flat)

          for i in range(f.n_out()):
            self.assertTrue(res_re[i].sparsity()==ref[i].sparsity())
            self.checkarray(res_re[i],ref[i])

          f.generate_out("test.txt",res)
          res2 = f.generate_in("test.txt")

          for i in range(f.n_out()):
            self.assertTrue(res2[i].sparsity()==ref[i].sparsity())
            self.checkarray(res2[i],ref[i])

      if ref_flat is None:
        with self.assertInException("mismatching shape"):
          res = f.nz_from_in(ins)

        with self.assertInException("mismatching shape"):
          f.generate_in("test.txt",ins)
      else:
        res = f.nz_from_in(ins)
        self.checkarray(res,ref_flat)

        in_re = f.nz_to_in(res)
        for i in range(f.n_in()):
          self.assertTrue(in_re[i].sparsity()==ref[i].sparsity())
          self.checkarray(in_re[i],ref[i]/2)

        f.generate_in("test.txt",ins)
        ins2 = f.generate_in("test.txt")

        for i in range(f.n_in()):
          self.assertTrue(ins2[i].sparsity()==ref[i].sparsity())
          self.checkarray(ins2[i],ref[i]/2)


    #Null dicts
    ins = {"i0": sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])), "i2": DM([1,3])}
    res = f.call(ins)

    self.checkarray(res["o0"],2*sparsify(DM([[1,0,0],[2,4,0],[7,8,9]])))
    self.checkarray(res["o1"],2*DM([2,2,2]))
    self.checkarray(res["o2"],2*DM([[1,1],[3,3]]))

    res = f.nz_from_in(f.convert_in(ins))
    self.checkarray(res,[1,2,7,4,8,9,2,2,2,1,3,1,3])

  def test_convert_in(self):
    x = MX.sym("x")
    y = MX.sym("y")
    z = MX.sym("z")
    f = Function("f",[x,y,z],[2*x,2*y,2*z,x+y],["x","y","z"],["a","b","c","d"],{"default_in": [1,2,3]})

    res = f.convert_in({"x":4})
    self.checkarray(res[0],4)
    self.checkarray(res[1],2)
    self.checkarray(res[2],3)

    with self.assertInException("xy"):
      f.convert_in({"xy":4})

    res = f.convert_in([7,3,4])
    self.checkarray(res["x"],7)
    self.checkarray(res["y"],3)
    self.checkarray(res["z"],4)

    with self.assertInException("Incorrect"):
      f.convert_in([7,3])

    res = f.convert_out({"b":4})
    self.checkarray(res[0],np.nan)
    self.checkarray(res[1],4)
    self.checkarray(res[2],np.nan)
    self.checkarray(res[3],np.nan)

    with self.assertInException("xy"):
      f.convert_out({"xy":4})

    res = f.convert_out([1,2,3,4])
    self.checkarray(res["a"],1)
    self.checkarray(res["b"],2)
    self.checkarray(res["c"],3)
    self.checkarray(res["d"],4)

    with self.assertInException("Incorrect"):
      f.convert_out([1,2,3])

  @requires_nlpsol("ipopt")
  def test_ad_weight_sp(self):
    x = MX.sym("x",4)

    f = Function('f',[x],[x])

    with self.assertOutput("1 forward sweeps needed","1 reverse sweeps needed"):
      jacobian(f(sin(x)),x,{"helper_options": {"verbose":True, "ad_weight_sp":0}})
    with self.assertOutput("1 reverse sweeps needed","1 forward sweeps needed"):
      jacobian(f(sin(x)),x,{"helper_options": {"verbose":True, "ad_weight_sp":1}})


    x = MX.sym("x",4)

    nlp = {"x":x,"f":dot(x,x),"g": sin(x)}

    with self.assertOutput("1 forward sweeps needed","1 reverse sweeps needed"):
      solver = nlpsol("solver","ipopt",nlp,{"common_options": {"helper_options": {"verbose":True, "ad_weight_sp":0}}})
    with self.assertOutput("1 reverse sweeps needed","1 forward sweeps needed"):
      solver = nlpsol("solver","ipopt",nlp,{"common_options": {"helper_options": {"verbose":True, "ad_weight_sp":1}}})


    f = Function('f',[x],[x],{"verbose":True,"ad_weight_sp":1})
    with self.assertOutput("1 reverse sweeps needed","1 forward sweeps needed"):
      jacobian(f(sin(x)),x)

    x = MX.sym("x",100)

    f = Function('f',[x],[x[0]],{"verbose":True,"ad_weight_sp":0})

    with self.assertOutput("2 forward sweeps needed","1 reverse sweeps needed"):
      jacobian(f(sin(x)),x)

  def test_map_exception(self):
    x = MX.sym("x",4)
    y = MX.sym("y",4)
    f = Function("f",[x],[x+y],{"allow_free":True})

    if "CASADI_WITH_THREAD" in CasadiMeta.compiler_flags():
      message = "Evaluation failed"
    else:
      message = "since variables [y] are free"

    with self.assertInException(message):
      F = f.map(4,"thread",2)
      F(3)
      
  def test_DM_arg(self):
    f = Function('f',[DM(0,1)],[])
    print(f)
    f = Function('f',[DM(0,1),SX.sym("x")],[])
    print(f)
    f = Function('f',[DM(0,1),MX.sym("x")],[])
    print(f)
    self.assertTrue(isinstance(vertcat(),DM))
    self.assertEqual(vertcat().shape,(0,1))
    self.assertTrue(isinstance(horzcat(),DM))
    self.assertEqual(horzcat().shape,(1,0))
    self.assertTrue(isinstance(veccat(),DM))
    self.assertEqual(veccat().shape,(0,1))
    self.assertTrue(isinstance(diagcat(),DM))
    self.assertEqual(diagcat().shape,(0,0))
    self.assertTrue(isinstance(vcat([]),DM))
    self.assertEqual(vcat([]).shape,(0,1))
    self.assertTrue(isinstance(hcat([]),DM))
    self.assertEqual(hcat([]).shape,(1,0))
    self.assertTrue(isinstance(vvcat([]),DM))
    self.assertEqual(vvcat([]).shape,(0,1))
    self.assertTrue(isinstance(dcat([]),DM))
    self.assertEqual(dcat([]).shape,(0,0))
    
    
  def test_nondiff(self):

    for X in [SX,MX]:

      x = X.sym("x",2,2)
      y = X.sym("y",2,2)

      xn = DM([[1,3],[0.1,2]])
      yn = DM([[4,5],[6,7]])

      options = {"is_diff_in":[True,False],"is_diff_out":[True,False]}
      f = Function("f",[x,y],[sin(x+y),x*y],options)



      F = Function("F",[x,y],f(cos(x),(x*y)**2),["x","y"],["z","zz"],options)


      for ff in [F.forward(1).forward(1),F.forward(1).reverse(1),F.reverse(1).forward(1),F.reverse(1).reverse(1)]:
        s = str(ff)
        self.assertTrue("y" not in s.replace("_y[2x2,0nz]","foo")[len("fwd1_adj1_F:(x[2x2],y[2x2]"):])


      ye = X(2,2);

      G = Function("G",[x,ye],[f(cos(x),(x*yn)**2)[0], f(cos(xn),(xn*yn)**2)[1]],["x","y"],["z","zz"])
      self.checkfunction(F,G,inputs=[xn,yn],evals=1)

      for f1 in [lambda f: f.forward(1), lambda f: f.reverse(1)]:
         Gf = f1(G)
         Ff = f1(F)

         arg = [xn,yn]+[DM.rand(Gf.sparsity_in(i)) for i in range(Gf.n_in())][2:]
         for a,b in zip(Gf.call(arg),Ff.call(arg)):
            self.checkarray(a,b)

      for f1 in [lambda f: f.forward(1), lambda f: f.reverse(1)]:
        for f2 in [lambda f: f.forward(1), lambda f: f.reverse(1)]:
           Gf = f1(f2(G))
           Ff = f1(f2(F))

           arg = [xn,yn]+[DM.rand(Gf.sparsity_in(i)) for i in range(Gf.n_in())][2:]
           for a,b in zip(Gf.call(arg),Ff.call(arg)):
             self.checkarray(a,b)

  def test_non_diff_call(self):
    x = MX.sym("x",5)
    y = MX.sym("x",5,5)

    f = Function("f",[x,y],[mtimes(y,x),mtimes(y.T,x)],{"never_inline":True,"is_diff_in":[False,True],"is_diff_out":[False,True]})
    
    

    x = SX.sym("x",5)
    y = SX.sym("x",5,5)
    
    g = Function('g',[x,y],f(x,y))
    

    X = MX.sym("x",5)
    Y = MX.sym("x",5,5)
    
    G = Function('g',[x,y],f(x,y))
    
    DM.rng(1)
    x0=DM.rand(5)
    y0=DM.rand(5,5)
    
    self.checkfunction(g,G,inputs=[x0,y0])
    

  @memory_heavy()
  def test_inline_linear_interpolant(self):

    do_inline = False

    np.random.seed(0)

    N = 3
    M = 4

    d_knots = [list(np.linspace(0,1,N)),list(np.linspace(0,1,M))]

    data0 = np.random.random([len(e) for e in d_knots])
    data1 = np.random.random([len(e) for e in d_knots])
    r = np.meshgrid(*d_knots,indexing='ij')

    xyz = np.vstack(list(e.ravel(order='F') for e in r)).ravel(order='F')

    d_flat0 = data0.ravel(order='F')
    d_flat1 = data1.ravel(order='F')

    data = np.vstack((data0.ravel(order='F'),data1.ravel(order='F'))).ravel(order='F')
    d_flat = data.ravel(order='F')


    LUT_param = casadi.interpolant('name','linear',[N,M],2, {"lookup_mode": ["exact","linear"]})

    only_fd = {"enable_fd": True, "enable_forward": False, "enable_reverse": False}
    options = {"ad_weight_sp":-1}
    options.update(only_fd)
    options["forward_options"] = only_fd
    options["reverse_options"] = only_fd
    #options["jacobian_options"] = only_fd

    print(options)
    LUT_param_ref = LUT_param.wrap_as_needed(options)
    J_ref = LUT_param_ref.jacobian()

    d_knots_cat = vcat(d_knots[0]+d_knots[1])

    inputs = [[0.2,0.333],d_knots_cat,d_flat]

    LUT_param = casadi.interpolant('name','linear',[N,M],2,{"inline": True, "lookup_mode": ["exact","linear"]})
    J = LUT_param.jacobian()

    self.checkarray(LUT_param(*inputs),LUT_param_ref(*inputs))
    inputs+= [0]
    print("J_ref",J_ref,J_ref(*inputs))
    self.checkfunction(J,J_ref,inputs=inputs,digits=8,digits_sens=1,evals=1)


  def test_issue3079(self):  
    interp1d([1,2,3], MX([4,5,6]), [1.1], 'linear')

  def test_functionbuffer(self):
    A_ = np.random.random((4,4))
    B_ = np.zeros((4,4))

    A = MX.sym("A",4,4)

    f = Function("f",[A],[A*10])


    [buf,f_eval] = f.buffer()
    buf.set_arg(0, memoryview(A_))
    buf.set_res(0, memoryview(B_))

    f_eval()

    self.checkarray(B_, 10*A_)

    self.assertEqual(buf.ret(), 0)

  @requires_conic("osqp")
  @requiresPlugin(Importer,"shell")
  def test_jit_buffer_eval(self):
    opti = Opti('conic')

    x = opti.variable()

    opti.minimize((x-3)**2)

    opti.subject_to(x>=0)

    opti.subject_to(x<=5)


    opti.solver('osqp')


    f = opti.to_function('f',[],[x])


    f.generate('f.c')

    libdir = GlobalOptions.getCasadiPath()
    includedir = GlobalOptions.getCasadiIncludePath()

    if os.name=='nt':
      f = opti.to_function('f',[],[x],{"jit":True,"compiler":"shell","jit_options":{"compiler_flags":["/I"+includedir],"linker_flags":["/LIBPATH:"+libdir,"osqp.lib"],"verbose":True}, "print_time": True})
    else:
      f = opti.to_function('f',[],[x],{"jit":True,"compiler":"shell","jit_options":{"compiler_flags":["-I"+includedir,"-L"+libdir,"-losqp"],"linker_flags":["-I"+includedir,"-L"+libdir,"-losqp"],"verbose":True}, "print_time": True})
    [buf,trigger] = f.buffer()

    a = np.array([1.0])
    buf.set_res(0, memoryview(a))

    trigger()

    # buffer jit eval bypasses timings
    with self.assertInException("No stats available"):
        f.stats()
    self.checkarray(a,3)

  def test_codegen_inf_nan(self):
    x = MX.sym("x")
    f = Function("F",[x],[x+inf])
    self.check_codegen(f,inputs=[1],std="c99",main=True)

    x = MX.sym("x")
    f = Function("F",[x],[x+np.nan])
    self.check_codegen(f,inputs=[1],std="c99",main=True)

    x = MX.sym("x")
    f = Function("F",[x],[x+vertcat(inf,np.nan,-inf)])
    self.check_codegen(f,inputs=[1],std="c99",main=True)

  def test_codegen_with_mem(self):
    x = MX.sym("x")
    f = Function("F",[x],[3*x])
    self.check_codegen(f,inputs=[1],main=True,opts={"with_mem":True},definitions=["inline=\"\""])
    self.check_codegen(f,inputs=[1],main=True,opts={"with_mem":True,"with_header":True},definitions=["inline=\"\""])

  def test_codegen_scalars_bug(self):
    x = MX.sym("x")
    z = 3*x/sin(x)
    f = Function("F",[x],[z,z/x],{"live_variables":False})
    self.check_codegen(f,inputs=[1],opts={"codegen_scalars":True})

  def test_bug_codegen_logical(self):
    a = MX([1,0,0])
    b = MX([1,1,0])
    c = logic_or(a,b)
    f = Function("f",[],[c])
    self.check_codegen(f,inputs=[])

  def test_jit_serialize(self):
    if not args.run_slow: return
    if sys.platform=="darwin": return

    def test_cases(jit_serialize):
      opts = {"jit":True, "compiler": "shell", "jit_options": {"verbose":True}, "verbose":False, "jit_serialize": jit_serialize}
      x = MX.sym("x")
      yield lambda : Function('f',[x],[(x-3)**2],opts)


      x = MX.sym("x", 2)
      f = x[0]*x[0] + x[1]*x[1]

      if has_nlpsol("ipopt"):
        yield lambda : nlpsol("solver", "ipopt", {"x": x, "f": f},opts)



    for case in test_cases("source"):

      with self.assertOutput(["jit_tmp"],[]):
        f = case()

      f.save('f.casadi')

      with self.assertOutput(["jit_tmp"],[]):
        g = Function.load('f.casadi')

      self.checkfunction_light(f, g, inputs={})

      g.save('f.casadi')
      with self.assertOutput(["jit_tmp"],[]):
        g = Function.load('f.casadi')

      self.checkfunction_light(f, g, inputs={})



    for case in test_cases("link"):

      with self.assertOutput(["jit_tmp"],[]):
        f = case()


      f.save('f.casadi')

      with self.assertOutput([],["jit_tmp"]):
        g = Function.load('f.casadi')

      self.checkfunction_light(f, g, inputs={})

      g.save('f.casadi')
      with self.assertOutput([],["jit_tmp"]):
        g = Function.load('f.casadi')

      self.checkfunction_light(f, g, inputs={})

      f = None
      g = None
      if os.name!='nt': # Workaround for known bug #3039
        with self.assertInException("No such file"):
          g = Function.load('f.casadi')


    for case in test_cases("embed"):

      with self.assertOutput(["jit_tmp"],[]):
        f = case()

      f.save('f.casadi')
      with self.assertOutput([],["jit_tmp"]):
        g = Function.load('f.casadi')
      g()
      f = None
      with self.assertOutput([],["jit_tmp"]):
        g = Function.load('f.casadi')

      g()
      g.save('f.casadi')
      with self.assertOutput([],["jit_tmp"]):
        g = Function.load('f.casadi')
      g()
      g = None
      with self.assertOutput([],["jit_tmp"]):
        g = Function.load('f.casadi')


  def test_map_get_function(self):
    x = MX.sym("x")

    g = Function("g",[x],[x**2])
    ff = g.map(5)

    self.assertTrue(len(ff.get_function())==1)
    f2 = ff.get_function("f")

    self.checkfunction_light(g, f2, inputs=[3])

  def test_post_expand(self):

    x = MX.sym("x")

    f = Function("f",[x],[x**2])

    print(f)

    f = Function("f",[x],[x**2],{"post_expand":True})
    print(f)
    self.assertTrue(f.is_a("SXFunction"))

    f = Function("f",[x],[x**2],{"post_expand":True,"post_expand_options":{"print_in":True}})
    print(f)
    self.assertTrue(f.is_a("SXFunction"))

    # Check that the option came through
    with self.assertOutput(["Input 0 (i0): 3"],[]):
      f(3)
  
  @requires_conic('osqp')
  def test_memful_main(self):
    c = conic("conic","osqp",{"a":Sparsity.dense(1,2),"h":Sparsity.dense(2,2)},{"print_problem":True,"osqp.verbose":False})
    inputs = {"h": DM([[1,0.2],[0.2,1]]),"g":vertcat(1,2),"a":horzcat(1,1),"lba":-1,"uba":1}
    inputs = c.convert_in(inputs)
    extra_options = ["-Wno-endif-labels","-Wno-unused-variable"]
    # No extra options on windows
    if os.name == 'nt':
      extra_options = []
    self.check_codegen(c,inputs=inputs,main=True,std="c99",extra_options=extra_options,extralibs=["osqp"])

  @requires_conic('osqp')
  def test_memful_external(self):
    if not args.run_slow: return
    c = conic("conic","osqp",{"a":Sparsity.dense(1,2),"h":Sparsity.dense(2,2)},{"print_problem":True,"osqp.verbose":False})
    inputs = {"h": DM([[1,0.2],[0.2,1]]),"g":vertcat(1,2),"a":horzcat(1,1),"lba":-1,"uba":1}
    inputs = c.convert_in(inputs)
    extra_options = ["-Wno-endif-labels","-Wno-unused-variable"]
    # No extra options on windows
    if os.name == 'nt':
      extra_options = []
    F,lib = self.check_codegen(c,inputs=inputs,std="c99",extra_options=extra_options,extralibs=["osqp"])
    F = F.wrap()
    print("memful_external")
    self.check_codegen(F,inputs=inputs,main=True,extralibs=[lib])
    
  def test_cse(self):
    for X in [SX,MX]:
        x = X.sym("x",2)
        y = X.sym("y",2)
        p = X.sym("p",2)
        z=p*cos(x+5*y)
        z2=p*cos(x+5*y)
        w = z-z2
        self.assertFalse(w.is_zero())
        res = cse(w)
        self.assertTrue(res.is_zero())
        f1 = Function('f',[x,y,p],[w],{"cse":False})
        f2 = Function('f',[x,y,p],[w],{"cse":True})
        self.assertTrue(f1.n_instructions()>3)
        self.assertTrue(f2.n_instructions()<=3)

  def test_cse_call(self):
    for X in [SX,MX]:
        x = X.sym("x")
        f = Function("f",[x],[x**2],["x"],["y"],{"never_inline":True})
        #fcopy = Function("f",[x],[x**2],["x"],["y"],{"never_inline":True})
        fcopy = f
        # Feature disabled, too expensive

        y = cse(vertcat(f(sin(x)),fcopy(sin(x)),f(sin(x))))
        
        print(y)
        
        if X is MX:
            self.assertTrue("vertcat(@1, @1, @1)" in str(y))
        else:
            self.assertTrue("[@1, @1, @1]" in str(y))

  @memory_heavy()
  def test_stop_diff(self):
    x = MX.sym("x")
    y = sin(x)
    
    max_order = 2
    for order in range(1, max_order+1):
    
        test_z = 3*x**(order-1)
        fa = Function('f',[x],[stop_diff(test_z,order)],['x'],['z'])
        fb = Function('f',[x],[test_z],['x'],['z'])
        self.checkfunction(fa, fb, inputs=[DM.rand(1,1)])
        
        z = stop_diff(y,order)
        DM.rng(1)
        
        v0 = DM.rand(1)#v.sparsity())
        expr0 = DM.rand(1)#expr.sparsity())
        
        for op in [gradient,jacobian,lambda expr,v : jtimes(expr,v,v0),lambda expr,v : jtimes(expr,v,expr0,True)]:
            yd = y
            zd = z
            for i in range(max_order+2):
                
                f = Function('f',[x],[yd,zd])
                x0 = DM.rand(f.sparsity_in(0))
                res = f(x0)
                

                if i>=order:
                    self.assertTrue(res[1].is_zero())
                else:
                    self.assertEqual(res[0],res[1])
        
                if i<order:
                    yd = op(yd,x)
                    zd = op(zd,x)
        
        f = Function('f',[x],[z],['x'],['z'])
        f_normal = Function('f',[x],[y],['x'],['z'])
        f_ref = Function('f',[x],[test_z],['x'],['z'])
        
        for i in range(3):
            inputs = [DM.rand(f.sparsity_in(i)) for i in range(f.n_in())]
            if i>=order:
                res = f.call(inputs,False,False)
                res_ref = f_ref.call(inputs,False,False)
                for e,e_ref in zip(res,res_ref):
                    if e_ref.is_zero():
                        self.assertTrue(e.is_zero())
                    else:
                        self.assertFalse(e.is_zero())
            else:
                self.checkfunction_light(f,f_normal,inputs=inputs)
                
            if i<order:
                
                in_labels = f.name_in()+['fwd:'+e for e in f.name_in()]+['adj:'+e for e in f.name_out()]
                out_labels = ['fwd:'+e for e in f.name_out()]+['adj:'+e for e in f.name_in()]
                for e_in in f.name_in():
                    for e_out in f.name_out():
                        out_labels.append('jac:%s:%s' % (e_out,e_in))
                        out_labels.append('grad:%s:%s' % (e_out,e_in))
                        
                f,f_normal,f_ref = [fe.factory('f',in_labels,out_labels) for fe in [f,f_normal,f_ref]]

  @memory_heavy()
  def test_stop_diff_cross(self):
    x = MX.sym("x",2)
    y = sin(x[0])+exp(x[1])+cos(x[0]**2*x[1]**2)
    
    max_order = 2
    for order in range(1, max_order+1):
    
        test_z = 3*x**(order-1)
        fa = Function('f',[x],[stop_diff(test_z,order)],['x'],['z'])
        fb = Function('f',[x],[test_z],['x'],['z'])
        self.checkfunction(fa, fb, inputs=[DM.rand(1,1)])
        
        z = stop_diff(y,order)
        DM.rng(1)

        for op in [jacobian,lambda expr,v : jtimes(expr,v,DM.rand(v.sparsity())),lambda expr,v : jtimes(expr,v,DM.rand(expr.sparsity()),True)]:
            yd = y
            zd = z
            for i in range(max_order+1):
                
                f = Function('f',[x],[yd,zd])
                x0 = DM.rand(f.sparsity_in(0))
                res = f(x0)
                

                if i>=order:
                    self.assertTrue(res[1].is_zero())
                else:
                    self.checkarray(res[0],res[1])
        
                if i<order:
                    DM.rng(5+i)
                    yd = op(yd,x)
                    DM.rng(5+i)
                    zd = op(zd,x)
        
        f = Function('f',[x],[z],['x'],['z'])
        f_normal = Function('f',[x],[y],['x'],['z'])
        f_ref = Function('f',[x],[test_z],['x'],['z'])
        
        if order<=1:
            for i in range(3):
                inputs = [DM.rand(f.sparsity_in(i)) for i in range(f.n_in())]
                if i>=order:
                    res = f.call(inputs,False,False)
                    res_ref = f_ref.call(inputs,False,False)
                    for e,e_ref in zip(res,res_ref):
                        if e_ref.is_zero():
                            self.assertTrue(e.is_zero())
                        else:
                            self.assertFalse(e.is_zero())
                else:
                    self.checkfunction_light(f,f_normal,inputs=inputs)
                    
                if i<order:
                    
                    in_labels = f.name_in()+['fwd:'+e for e in f.name_in()]+['adj:'+e for e in f.name_out()]
                    out_labels = ['fwd:'+e for e in f.name_out()]+['adj:'+e for e in f.name_in()]
                    for e_in in f.name_in():
                        for e_out in f.name_out():
                            out_labels.append('jac:%s:%s' % (e_out,e_in))
                            
                    f,f_normal,f_ref = [fe.factory('f',in_labels,out_labels) for fe in [f,f_normal,f_ref]]

  @memory_heavy()
  def test_stop_diff_p(self):
    # Drop this -> not implemented
    x = MX.sym("x")
    y = MX.sym("y")
    yy = sin(y*x)
    
    max_order = 2
    
    for order in range(1, max_order+1):
        DM.rng(1)
        print("order",order)
        test_z = 3*(y*x)**(order-1)+y**order
        test_z = sin(y)*x**(order-1)
        """fa = Function('f',[x,p],[stop_diff(test_z,x,order)],['x','p'],['z'])
        fb = Function('f',[x,p],[test_z],['x','p'],['z'])
        print()
        self.checkfunction(fa, fb, inputs=[DM.rand(1,1),DM.rand(1,1)])
        continue"""
        
        z = stop_diff(yy,x,order)

        
        v0 = DM.rand(1)#v.sparsity())
        expr0 = DM.rand(1)#expr.sparsity())
        
        
        for op in [gradient,jacobian,lambda expr,v : jtimes(expr,v,v0),lambda expr,v : jtimes(expr,v,expr0,True)]:
            yd = yy
            zd = z
            for i in range(max_order+2):
                
                f = Function('f',[x,y],[yd,zd])
                x0 = DM.rand(f.sparsity_in(0))
                p0 = DM.rand(f.sparsity_in(1))
                res = f(x0, p0)
                

                if i>=order:
                    self.assertTrue(res[1].is_zero())
                else:
                    self.assertEqual(res[0],res[1])
        
                if i<order:
                    yd = op(yd,x)
                    zd = op(zd,x)
        
        f = Function('f',[x,y],[z],['x','y'],['z'])
        f_normal = Function('f',[x,y],[yy],['x','y'],['z'])
        f_ref = Function('f',[x,y],[test_z],['x','y'],['z'])
        
        for i in range(3):
            inputs = [DM.rand(f.sparsity_in(i)) for i in range(f.n_in())]
            if i>=order:
                res = f.call(inputs,False,False)
                res_ref = f_ref.call(inputs,False,False)
                f_ref.disp(True)
                f.generate('f.c')
                for e,e_ref,name in zip(res,res_ref,f.name_out()):
                    print("check",name,e,e_ref)
                    if e_ref.is_zero():
                        self.assertTrue(e.is_zero())
                    else:
                        self.assertFalse(e.is_zero())
            else:
                self.checkfunction_light(f,f_normal,inputs=inputs)
                
            if i<order:
                
                in_labels = f.name_in()+['fwd:'+e for e in f.name_in()]*0+['adj:'+e for e in f.name_out()]
                out_labels = f.name_out()+['fwd:'+e for e in f.name_out()]*0+['adj:'+e for e in f.name_in()]
                #for e_in in f.name_in():
                #    for e_out in f.name_out():
                #        out_labels.append('jac:%s:%s' % (e_out,e_in))
                #        out_labels.append('grad:%s:%s' % (e_out,e_in))
                        
                f,f_normal,f_ref = [fe.factory('f',in_labels,out_labels) for fe in [f,f_normal,f_ref]]
            
        
  @requires_conic("qpoases")
  def test_no_hess2(self):
    y = MX.sym("y",2)

    f = Function("foo",[y],[vertcat(sin(y[0]*y[1]),cos(y[0]*y[1]))],{"never_inline":True,"print_out":True})


    opti = Opti()

    x = opti.variable(2)

    opti.subject_to(no_hess(f(x**2))>=0)

    opti.minimize(sumsqr(x-2))

    opti.solver("sqpmethod")

    F = opti.to_function("F",[x],[x])
    F.generate('F.c')
    with open('F.c','r') as inp:
        code = inp.read()
    for m in re.findall(r"\w+FS",code):
        self.assertTrue(len(m.split("_"))<=2)
    for m in re.findall(r"\w+foo",code):
        pass
        # bug 3019
        #self.assertTrue(len(m.split("_"))<=2)

  def test_issue_1522(self):
    V2 = MX.sym("X",8)
    x,travel = vertsplit(V2,[0,6,8])
    xs = vertsplit(x,[0,3,6])
    travels = vertsplit(travel,[0,1,2])

    dist = 0

    for j in range(2):
      dist+=sum1((xs[0]-(xs[j]+travels[j]))**2)

    nlp = {"x":V2,"f":-dist}


    p = MX(0,1)
    f = Function('f',[V2,p],[-dist])


    for compact in [True,False]:
        for F in [f, f.expand()]:
            HF = F.reverse(1).jac_sparsity(0,0,compact)
            self.assertTrue(HF.is_symmetric())

  def test_issue_3074(self):
    x = MX.sym("x",4)
    y = x[:2]
    y = 1
    f = Function('f',[x],[dot(y,y)])
    print(f)
    fr = f.reverse(1)
    print(fr)
    #print(fr.jac_sparsity())
  
    x = MX.sym("x",sparsify(DM([0,1,1,0])).sparsity())
    y = dot(x,x)
    f = Function('f',[x],[dot(y,y)])
    print(f)
    fr = f.reverse(1)
    print(fr)
    print(fr.jac_sparsity())
    
    data = "jhpnnagiieahaaaadaaaaaaaaaaaaaaaaafaegaakaaaaaaaneifgefhogdgehjgpgogbaaaaaaapaaaaaaaegfgmgbgjhpfbgchhgfhngfgogehdhaaaaaacaaaaaaahaaaaaaaaaaaaaaabababababababaaaaaaaaaaaaaaaaahaaaaaaaaaaaaaaaegfaaaaaaaaaaaaaaabaaaaaaaaaaaaaaabaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaabaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaeggaaaaaaaaaaaaaaacaaaaaaaaaaaaaaabaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaacaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaabaaaaaaaaaaaaaaachbaaaaaaaaaaaaaaaegkaaaaaaaaaaaaaaagaaaaaaaaaaaaaaabaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaagaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaabaaaaaaaaaaaaaaacaaaaaaaaaaaaaaadaaaaaaaaaaaaaaaeaaaaaaaaaaaaaaafaaaaaaaaaaaaaaaegeaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaabaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaachdaaaaaaaaaaaaaaaegiaaaaaaaaaaaaaaaeaaaaaaaaaaaaaaabaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaeaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaabaaaaaaaaaaaaaaacaaaaaaaaaaaaaaadaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaahaaaaaaaaaaaaaaacaaaaaaajgadcaaaaaaajgbdcaaaaaaajgcdcaaaaaaajgddcaaaaaaajgedcaaaaaaajgfdcaaaaaaajggdaaaaaaaaaaaaaaaaaabagaaaaaaadhpgfhchdgfgbahaaaaaaakgjgehpfehngahaaaaaaaaaaaaaaaafaaaaaaadgmgbgoghgaaegbaaaaaaaaaaaaaaaaebababaaabababaaapbfilobfilobfnpdmfpicmfpicmfpnpdaaaaaeaaaaaaaaaaaaaaaabakdmiadcooijhfeodaaaaaaaaaaaaaaaabaaaaaaaocdaaaaaaangehihaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaachfaaaaaaaaaaaaaaahaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaabaaaaaaahaaaaaaaaaaaaaaaegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaaeaaaaaaaehjgngfgegndaaaaaacaaaaaaaaaaaaaaaegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaagaaaaaaacgpgegjhocihegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaaiaaaaaaacgpgegjhocghpfihchbaaaaaaaaaaaaaaaegndaaaaaacaaaaaaaaaaaaaaaegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaalaaaaaaaegfgchiccgpgegjhocihjcegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaanaaaaaaaegfgchiccgpgegjhocghpfihjcchbaaaaaaaaaaaaaaaegndaaaaaagaaaaaaaaaaaaaaaegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaaiaaaaaaacgpgegjhocggpfihegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaaiaaaaaaacgpgegjhocbgpfihegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaajaaaaaaabgdgdgfgmgocbgpfihegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaakaaaaaaabgdgdgfgmgocngbgpfihegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaalaaaaaaabgdgdgfgmgoccgpfihocfhegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaalaaaaaaabgdgdgfgmgoccgpfihocjhchcaaaaaaaaaaaaaaaegmcaaaaaaadaaaaaaaaaaaaaaaachdaaaaaaaaaaaaaaaegmcaaaaaaadaaaaaaaaaaaaaaaachdaaaaaaaaaaaaaaaegndaaaaaaeaaaaaaaaaaaaaaaegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaagaaaaaaacgpgegjhochgegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaagaaaaaaacgpgegjhocdgegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaagaaaaaaacgpgegjhocngegpcaaaaaaaaaaaaaaaaaaaaaachaaaaaaaaaaaaaaaalaaaaaaabgdgdgfgmgoccgpfihoccgcheaaaaaaaaaaaaaaabaaaaaaaaaaaaaaaaaaaaaaabaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaahaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaabaaaaaaaaaaaaaaaaa"


    f = StringDeserializer(data).unpack()
    print(f.reverse(1))
    f.reverse(1).jac_sparsity()
   
  @requires_nlpsol("ipopt")
  def test_issue_3134(self):
    p = MX.sym("p")
    v = MX.sym("v")

    u = MX.sym("u")
    x = vertcat(p,v)

    rhs = vertcat(v,u)

    t0 = MX.sym("t0")
    DT = MX.sym("DT")

    F = Function('F', [x, u, t0, DT], [x + DT * rhs], ['x0', 'u', 't0', 'DT'], ['xf'])


    x = []
    g = []

    Xk = MX.sym("Xk",2)
    x.append(Xk)

    T = MX.sym("T")
    x.append(T)

    g.append(T)

    t0 = MX.sym("t0")
    x.append(t0)

    # Note: (t0+T)-(t0+T/2) does not simplify; sparsity pattern indicates dependence on T and t0
    DTs = [T/2,(t0+T)-(t0+T/2)]

    for i in range(2):
        Uk = MX.sym("Uk")
        x.append(Uk)
        Xk_next = MX.sym("Xk_next",2)
        x.append(Xk_next)
        Xf = F(x0=Xk, u=Uk, t0=t0, DT=DTs[i])["xf"]
        g.append(Xk_next-Xf)
        Xk = Xk_next

    nlp = {"x": vvcat(x),"f": T, "g": vvcat(g)}

    solver = nlpsol("solver","ipopt",nlp)
   
  def test_which_depends(self):
    x = SX.sym("x",2)
    y = SX.sym("y",2)
    
    z = x+y
    z2 = x*y
    z3 = sin(dot(x,y))
    z4 = cos(x)
    z5 = sin(y)
    z6 = 12
    z7 = vertcat(x[0],y[1])
    z8 = vertcat(x[0]**2,y[1]**2)
    
    f = Function("f",[x,y],[z,z2,z3,z4,z5,z6,z7,z8],["x","y"],["z","z2","z3","z4","z5","z6","z7","z8"])
    
    def mychecks(f,exempt=False):
        self.assertEqual(f.which_depends("x",["z"]),[True,True])
        self.assertEqual(f.which_depends("x",["z"],2),[False,False])
        if not exempt:
            self.assertEqual(f.which_depends("x",["z2"],2),[False,False])
        self.assertEqual(f.which_depends("x",["z3"],2),[True,True])
        self.assertEqual(f.which_depends("x",["z4"]),[True,True])
        self.assertEqual(f.which_depends("x",["z4"],2),[True,True])
        self.assertEqual(f.which_depends("x",["z5"]),[False,False])
        self.assertEqual(f.which_depends("x",["z5"],2),[False,False])
        self.assertEqual(f.which_depends("x",["z5","z6"]),[False,False])
        if not exempt:
            self.assertEqual(f.which_depends("x",["z","z2","z3","z4","z5","z6"],1,True),[True, True]+[True, True]+ [True]+[True,True]+[False, False]+[False])
            self.assertEqual(f.which_depends("x",["z","z2","z3","z4","z5","z6"],2,True),[False, False]+[False, False]+ [True]+[True,True]+[False, False]+[False])
        self.assertEqual(f.which_depends("x",["z","z3","z4","z5","z6"],1,True),[True, True]+ [True]+[True,True]+[False, False]+[False])
        self.assertEqual(f.which_depends("x",["z","z3","z4","z5","z6"],2,True),[False, False]+ [True]+[True,True]+[False, False]+[False])
        self.assertEqual(f.which_depends("x",["z7"]),[True,False])
        self.assertEqual(f.which_depends("y",["z7"]),[False,True])
        self.assertEqual(f.which_depends("x",["z7"],2),[False,False])
        self.assertEqual(f.which_depends("y",["z7"],2),[False,False])
        self.assertEqual(f.which_depends("x",["z8"],2),[True,False])
        self.assertEqual(f.which_depends("y",["z8"],2),[False,True])
        self.assertEqual(f.which_depends("x",["z7","z8"],1,True),[True, False]+[True, False])
        self.assertEqual(f.which_depends("x",["z7","z8"],2,True),[False, False]+[True, False])
        self.assertEqual(f.which_depends("y",["z7","z8"],1,True),[False, True]+[False, True])
        self.assertEqual(f.which_depends("y",["z7","z8"],2,True),[False, False]+[False, True])
    
    mychecks(f)
    
    if not args.run_slow: return
    F,_ = self.check_codegen(f,inputs=[DM.rand(f.sparsity_in(i)) for i in range(f.n_in())],with_jac_sparsity=True,with_forward=True)
    mychecks(F,exempt=True)
   
   
  def test_cat_input(self):

    x = MX.sym("x",2)
    
    fmx = Function('f',[vertcat(MX(2,1),x)],[sin(x)],{"always_inline":True})
    print(fmx)
    
    X = MX.sym("X",4)#Sparsity.lower(3))
    print(fmx(X))
    
    f_eval = Function('f_eval',[X],[fmx(X)])
    print(f_eval([np.nan,np.nan,3,np.nan]))
    

  def test_reshape_input(self):

    x = MX.sym("x",6,2)
    xsx = SX.sym("x",6,2)
    
    fmx = Function('f',[reshape(x,3,4)],[sin(x)],{"always_inline":True})
    fsx = Function('f',[reshape(xsx,3,4)],[sin(xsx)])
    
    DM.rng(1)
    inputs = [DM.rand(3,4)]
    for input in inputs:
        self.checkfunction_light(fmx,fsx,inputs=[input])
    
 
 
  def test_sparsity_cast_input(self):
  
    X = MX.sym("X",Sparsity.lower(3))
    f = Function("f",[X],[X**2],{"always_inline":True})
  
    sp = Sparsity.lower(3)
    N = sp.nnz()
    
    x = MX.sym("x",N)
    xsx = SX.sym("x",N)
    
    fmx = Function('f',[sparsity_cast(x,sp)],[sin(x)],{"always_inline":True})
    fsx = Function('f',[sparsity_cast(xsx,sp)],[sin(xsx)])
    
    inputs = [sparsify(DM([[1,0,0],[3,7,0],[1,8,9]])), sparsify(DM([[1,0,0],[3,0,0],[0,8,9]])), DM([[1,1.1,8],[3,1.2,3],[1,8,9]])]
    for input in inputs:
        self.checkfunction_light(fmx,fsx,inputs=[input])
    
    for X in [MX.sym("X",Sparsity.lower(3)), MX.sym("X",Sparsity.diag(3)), MX.sym("X",3,3)]:
        fcmx = Function('f',[X],[fmx(X)])
        fcsx = Function('f',[X],[fsx(X)])
        
        self.checkfunction_light(fcmx,fcsx,inputs=[DM.rand(X.sparsity())])

    
  def test_external(self):
    if not args.run_slow: return
    with self.assertInException("config failed"):
        self.compile_external("F","assets/externalfun1.c")
    
    for n_args in [3,7]:
        [externalfun1,libname] = self.compile_external("F","assets/externalfun1.c",{"config_args":["foo"]*n_args},debug_mode=True)
        print(libname)
        r = externalfun1(0)
        print(externalfun1.stats())
        self.assertEqual(n_args+1,int(r[0]))
        self.assertEqual(0,int(r[1]))

        x = MX.sym("x")
        
        g = Function('g',[x],[    externalfun1(sin(x))[0]+externalfun1(cos(x))[0] ])
        
        self.assertEqual(g(0.3),2*(n_args+1))
        self.check_codegen(g,inputs=[0.3],extralibs=[libname])
        
        self.check_serialize(externalfun1,inputs=[0.3])
        self.check_serialize(g,inputs=[0.3])
        
        externalfun1 = None
        g = None
        import gc
        gc.collect()
        
  def test_cache(self):
    x = MX.sym("x")
    f = Function('f',[x],[x**2])
    
    ff = f.forward(1)
    ff2 = f.forward(1)
    
    h = hash(ff)
    
    self.assertEqual(hash(ff),hash(ff2))
    
    ff = None
    ff2 = None
    
    import gc
    gc.collect()
    
    f2a = f.forward(2)
    f2b = f.forward(2)
    
    # Should be recreated because it was purged
    ff = f.forward(1)
    #self.assertNotEqual(hash(ff),h) # does not pass on Windows, but this test looks unreliable anyway
    self.assertEqual(hash(f2a),hash(f2b))
    
  def test_copy_elision(self):
    import casadi as ca
    
    backup = ca.GlobalOptions.getCopyElisionMinSize()
    
    for enabled in [False,True]:
    
      ca.GlobalOptions.setCopyElisionMinSize(2 if enabled else -1)

      x = ca.MX.sym("x",10)

      a = ca.MX.sym("a",5)

      bs = ca.MX.sym("b",10)

      b = (12*bs).monitor("b")
      y = 6*x


      [s0,s1] = vertsplit(x,[0,7,10])
      [ss0,ss1] = vertsplit(sin(x),[0,5,10])

      z = y[2:7]+a+b[2:7]
      z = sum1(z)
      z2 = vertcat(x,a,b)*9

      f = ca.Function('f',[x,a,bs],[z,z2, x, sqrt(s1), ss0+ss1])

      f.disp(True)

      f.generate('f.c')
      
      code = open('f.c','r').read()
      
      self.assertEqual("wr3" in code, enabled)
      
      lines = """
  casadi_int i, j, k;
  casadi_real *rr, w0, *w1=w+2, w2, *w3=w+8, *w4=w+18, *w5=w+28, *w6=w+33, *w7=w+38, *w8=w+43, *w9=w+68;
  const casadi_real *cs, *ct, *wr3, *wr4, *wr6, *wr9;
  /* #0: @0 = 0 */
  w0 = 0.;
  /* #1: @1 = ones(1x5) */
  casadi_fill(w1, 5, 1.);
  /* #2: @2 = 6 */
  w2 = 6.;
  /* #3: @3 = input[0][0] */
  wr3 = arg[0] ? arg[0] : casadi_zeros;
  /* #4: @4 = (@2*@3) */
  for (i=0, rr=w4, cs=wr3; i<10; ++i) (*rr++)  = (w2*(*cs++));
  /* #5: @5 = @4[2:7] */
  for (rr=w5, cs=w4+2; cs!=w4+7; cs+=1) *rr++ = *cs;
  /* #6: @6 = input[1][0] */
  wr6 = arg[1] ? arg[1] : casadi_zeros;
  /* #7: @5 = (@5+@6) */
  for (i=0, rr=w5, cs=wr6; i<5; ++i) (*rr++) += (*cs++);
  /* #8: @2 = 12 */
  w2 = 12.;
  /* #9: @4 = input[2][0] */
  wr4 = arg[2] ? arg[2] : casadi_zeros;
  /* #10: @4 = (@2*@4) */
  for (i=0, rr=w4, cs=wr4; i<10; ++i) (*rr++)  = (w2*(*cs++));
  /* #11: @4 = monitor(@4, b) */
  CASADI_PRINTF("b:\\n");
  casadi_print_canonical(casadi_s0, w4);
  CASADI_PRINTF("\\n");
  /* #12: @7 = @4[2:7] */
  for (rr=w7, cs=w4+2; cs!=w4+7; cs+=1) *rr++ = *cs;
  /* #13: @5 = (@5+@7) */
  for (i=0, rr=w5, cs=w7; i<5; ++i) (*rr++) += (*cs++);
  /* #14: @0 = mac(@1,@5,@0) */
  for (i=0, rr=(&w0); i<1; ++i) for (j=0; j<1; ++j, ++rr) for (k=0, cs=w1+j, ct=w5+i*5; k<5; ++k) *rr += cs[k*1]**ct++;
  /* #15: output[0][0] = @0 */
  if (res[0]) res[0][0] = w0;
  /* #16: @0 = 9 */
  w0 = 9.;
  /* #17: @8 = vertcat(@3, @6, @4) */
  rr=w8;
  for (i=0, cs=wr3; i<10; ++i) *rr++ = *cs++;
  for (i=0, cs=wr6; i<5; ++i) *rr++ = *cs++;
  for (i=0, cs=w4; i<10; ++i) *rr++ = *cs++;
  /* #18: @8 = (@0*@8) */
  for (i=0, rr=w8, cs=w8; i<25; ++i) (*rr++)  = (w0*(*cs++));
  /* #19: output[1][0] = @8 */
  casadi_copy(w8, 25, res[1]);
  /* #20: output[2][0] = @3 */
  casadi_copy(wr3, 10, res[2]);
  /* #21: {NULL, @9} = vertsplit(@3) */
  wr9 = wr3+7;
  /* #22: @9 = sqrt(@9) */
  for (i=0, rr=w9, cs=wr9; i<3; ++i) *rr++ = sqrt( *cs++ );
  /* #23: output[3][0] = @9 */
  casadi_copy(w9, 3, res[3]);
  /* #24: @3 = sin(@3) */
  for (i=0, rr=w3, cs=wr3; i<10; ++i) *rr++ = sin( *cs++ );
  /* #25: {@6, @1} = vertsplit(@3) */
  casadi_copy(w3, 5, w6);
  casadi_copy(w3+5, 5, w1);
  /* #26: @6 = (@6+@1) */
  for (i=0, rr=w6, cs=w1; i<5; ++i) (*rr++) += (*cs++);
  /* #27: output[4][0] = @6 */
  casadi_copy(w6, 5, res[4]);
  return 0;
"""
      if enabled:
        print(code)
        print(lines.strip())
        self.assertTrue(lines.strip() in code)
      self.check_codegen(f,inputs=[DM.rand(f.sparsity_in(i)) for i in range(f.n_in())])
      
      x = MX.sym("x",100,10)

      i = MX.sym("i")
      y = x[4:90,i]
      f = Function("f",[i,x],[sin(y)+x[7,i]])

      f.disp(True)

      f.generate('f.c')
      
      code = open('f.c','r').read()
      
      self.assertEqual("wr" in code, enabled)
      
      lines = """
  casadi_int i;
  casadi_real *rr, w1, w2, w3, *w4=w+1003;
  const casadi_real *cs, *wr0, *wr4;
  /* #0: @0 = input[1][0] */
  wr0 = arg[1] ? arg[1] : casadi_zeros;
  /* #1: @1 = 100 */
  w1 = 100.;
  /* #2: @2 = input[0][0] */
  w2 = arg[0] ? arg[0][0] : 0;
  /* #3: @3 = floor(@2) */
  w3 = floor( w2 );
  /* #4: @1 = (@1*@3) */
  w1 *= w3;
  /* #5: @4 = @0[(4:90;@1)] */
  wr4 = wr0 + 4 + (int)w1;
  /* #6: @4 = sin(@4) */
  for (i=0, rr=w4, cs=wr4; i<86; ++i) *rr++ = sin( *cs++ );
  /* #7: @1 = 100 */
  w1 = 100.;
  /* #8: @2 = floor(@2) */
  w2 = floor( w2 );
  /* #9: @1 = (@1*@2) */
  w1 *= w2;
  /* #10: @2 = @0[(7;@1)] */
  i = 7 + (int) w1;
  w2 = i>=0 && i<1000 ? wr0[i] : casadi_nan;
  /* #11: @4 = (@4+@2) */
  for (i=0, rr=w4; i<86; ++i) (*rr++) += w2;
  /* #12: output[0][0] = @4 */
  casadi_copy(w4, 86, res[0]);
  return 0;
"""
      if enabled:
        print(code)
        print(lines.strip())
        self.assertTrue(lines.strip() in code)
      self.check_codegen(f,inputs=[3,DM.rand(f.sparsity_in(1))],std="c99")
    ca.GlobalOptions.setCopyElisionMinSize(backup)
    
  @skip("simde" not in CasadiMeta.feature_list())
  @memory_heavy()
  @requiresPlugin(Importer,"shell")
  def test_blazing_spline(self):
    for N in [1,2,3]:
      
        def knots_expand(k):
            return [k[0]]*3 + k + [k[-1]]*3
      
        knots0 = [0,0.2,0.5,0.8,1]
        knots1 = [0,0.1,0.5,0.9,1]
        knots2 = [0,0.4,1]
        
        x0 = [0.4,0.5,0.6][:N]
        
        knots_orig = [knots0, knots1, knots2][:N]
        knots = [knots_expand(e) for e in knots_orig][:N]
        
        
        x=MX.sym("x",N)
        
        nc = np.prod([len(k)-4 for k in knots])
        DM.rng(1)
        data = DM.rand(nc)
        C = MX.sym("C",nc,1)
        Y = bspline(x,C,knots,[3]*N,1)
        F_ref = Function('f',[x,C],[Y])
        
        print(Y)
        
        F_ref2 = Function('f',[x,C],[Y,jacobian(Y,x)])
        
        F_ref2([0]*N,data)
        
        print(knots)
        
        
        if os.name=='nt':
            F = blazing_spline("F",knots,{"jit":True,"jit_options":{"flags": ["/I"+GlobalOptions.getCasadiIncludePath()]}})
        else:
            F = blazing_spline("F",knots,{"jit":True,"jit_options":{"flags": ["-I"+GlobalOptions.getCasadiIncludePath(),"-g","-ffast-math","-march=native"]}})
        
        def test_points(knots):
            import itertools
            selector = [lambda e: e[0]-0.1,lambda e: e[0],lambda e: (e[0]+e[1])/2,lambda e: e[-1],lambda e: e[-1]+0.1]
            for s in itertools.product(selector,repeat=len(knots)):
                yield [e(k) for e,k in zip(s,knots_orig)]

        for a in test_points(knots):
            self.checkfunction_light(F,F_ref,inputs=[vcat(a),data])
            self.check_serialize(F,inputs=[vcat(a),data])
       
        y = sin(F(x,C))
        
        f = Function('f',[x,C],[y,jacobian(y,x),gradient(y,x)]+list(hessian(y,x)))
        fcse = Function('fcse',[x,C],[y,jacobian(y,x),gradient(y,x)]+list(hessian(y,x)),{"cse":True})
        
        with capture_stdout() as result:
            fcse.disp(True)
            
        self.assertEqual(result[0].count(" F_der_der("),1)
        self.assertEqual(result[0].count(" F_der("),0)
        self.assertEqual(result[0].count(" F("),0)
        self.checkfunction_light(f,fcse,inputs=[vcat(x0),data])
        f = Function('f',[x,C],[y,jacobian(y,x),gradient(y,x)])
        fcse = Function('fcse',[x,C],[y,jacobian(y,x),gradient(y,x)],{"cse":True})
        
        with capture_stdout() as result:
            fcse.disp(True)
        self.assertEqual(result[0].count(" F_der_der("),0)
        self.assertEqual(result[0].count(" F_der("),1)
        self.assertEqual(result[0].count(" F("),0)
        fcse.disp(True)
        self.checkfunction_light(f,fcse,inputs=[vcat(x0),data])
        


        FJ = F.jacobian()
        FJ.generate("FJ.c",{"main":True})
        FJ.generate_in("FJ_in.txt",[vcat(a),data,0])
        FJ_ref = F_ref.jacobian()
        
        print("ref")
        F_ref2([0]*N,data)
        
        for a in test_points(knots):
            self.checkfunction_light(FJ,FJ_ref,inputs=[vcat(a),data,0])
       
        FJ = FJ.jacobian()
        FJ.generate("FH.c",{"main":True})
        FJ.generate_in("FH_in.txt",[vcat(x0),data,0,0,0])
        print(FJ)

        FJ_ref = FJ_ref.jacobian()
        import time
        t0 = time.time()
        FJ_ref(vcat(a),data,0,0,0)
        print("FJ_ref" ,time.time()-t0)
        t0 = time.time()
        FJ(vcat(a),data,0,0,0)
        print("FJ" ,time.time()-t0)
        for a in test_points(knots):
            print(vcat(a),data,0,0,0)
            self.checkfunction_light(FJ,FJ_ref,inputs=[vcat(a),data,0,0,0])
            
            
  def test_noncanonical_sparsity(self):
    x = MX.sym("x",4,4)
    y = MX.sym("y")
    
    f = Function("f",[x,y],[3*8])
    
    self.check_codegen(f,inputs=[DM.rand(4,4),1],opts={"force_canonical":False})
    self.check_codegen(f,inputs=[DM.rand(4,4),1],opts={"force_canonical":True})
    
  def test_options_sanitize(self):
      def canonical(e):
        if isinstance(e,dict):
            return [(k,canonical(e[k])) for k in sorted(e.keys())]
        return e
      def cmp(a,b):
        assert str(canonical(a))==str(canonical(b))
      res = Options.sanitize({"foo": {"bar": {"baz.goot": 7}}})
      ref = {"foo":{"bar": {"baz": {"goot":7}}}}
      cmp(res,ref)
      res = Options.sanitize({"foo.baz": 1, "foo":{"bar": 9}})
      ref = {'foo': {'bar': 9, 'baz': 1}}
      cmp(res,ref)
      res = Options.sanitize({"foo.baz.bar": 1, "foo":{"baz": {"go":8}}})
      ref = {"foo":{"baz":{"bar": 1, "go": 8}}}
      cmp(res,ref)
      res = Options.sanitize({"foo":{"bar": 9}, "foo.baz": 1})
      ref = {"foo":{"bar":9, "baz":1}}
      cmp(res,ref)
      res = Options.sanitize({"foo.bar.baz": 1, "foo.bar.mat":3,"foo.lib":5, "foo.bar.qint.pad":5,"foo":{"bar": {"got": 9}},"bar":{"abc.def":7}})
      ref = {'bar': {'abc': {'def': 7}}, 'foo': {'bar': {'baz': 1, 'mat': 3, 'got': 9, 'qint': {'pad': 5}}, 'lib': 5}}
      cmp(res,ref)
      
      with self.assertInException("update_dict error"):
          res = Options.sanitize({"foo": 9, "foo.baz": 1})
      with self.assertInException("update_dict error"):
          res = Options.sanitize({"foo": {"baz":{"w": 5}}, "foo.baz": 1})
          
      res = Options.sanitize({"foo": 1, "bar": None})
      print(res)
      ref = {"foo": 1}
      cmp(res,ref)
      # It is important that None/null entries are preserved deeper than toplevel
      # Use for augmented_options machinery in integrator
      res = Options.sanitize({"foo": 1, "bar": {"r": None}})
      print(res)
      ref = {"foo": 1, "bar": {"r": None}}
      cmp(res,ref)
if __name__ == '__main__':
    unittest.main()   
