misc/assertion.py

This example looks at the printme method

View output (PDF) | source (python)

See also

casadi::MX::attachAssert();
casadi::Callback();

 
from casadi import *
 


 

 
x = MX.sym("x")
y = sin(x)
z = sqrt(y)
 
f = Function("f", [x], [z])
 
z0 = f(5)
 
print(z0)
 

 

 
y = y.attachAssert(y>0, "bummer") 
z = sqrt(y)
 
f = Function("f", [x],[z])
 
try:
  z0 = f(5)
except Exception as e:
  print("An exception was raised here:")
  print(e)
 
 

 
class Dummy(Callback):
  def __init__(self, name, opts={}):
    Callback.__init__(self)
    self.construct(name, opts)
  def get_n_in(self): return 1
  def get_n_out(self): return 1
  def eval(self, arg):
    import numpy
    x = arg[0]
    m = max(numpy.real(numpy.linalg.eig(blockcat([[x,-1],[-1,2]]))[0]))
    print("m=",m)
    return [int(m>2)]
 
foo = Dummy("foo")
 
y = sin(x)
 
y = y.attachAssert(foo(y), "you are in trouble") 
z = sqrt(y)
 
f = Function("f", [x],[z])
 
z0 = f(5)