示例#1
0
文件: sparsity.py 项目: kozatt/casadi
 def test_intersection(self):
   self.message("Sparsity intersection")
   nza = set([  (0,0),
            (0,1),
            (2,0),
            (3,1),
            (2,3)])
   nzb = set([  (0,2),
            (0,0),
            (2,2),
            (2,3)])
   
   a = CRSSparsity(4,5)
   for i in nza:
     a.getNZ(i[0],i[1])
     
   b = CRSSparsity(4,5)  
   for i in nzb:
     b.getNZ(i[0],i[1])
   
   w = UCharVector()
   c=a.patternUnion(b,w,True,True,True)
   for k in range(c.size()):
     ind = (c.getRow()[k],c.col(k))
     self.assertTrue(ind in nza and ind in nzb)
       
   c = a * b
   self.assertEquals(c.size(),len(nza.intersection(nzb)))
   for k in range(c.size()):
     ind = (c.getRow()[k],c.col(k))
     self.assertTrue(ind in nza and ind in nzb)
示例#2
0
文件: sparsity.py 项目: kozatt/casadi
 def test_union(self):
   self.message("Sparsity union")
   nza = set([  (0,0),
            (0,1),
            (2,0),
            (3,1)])
   nzb = set([  (0,2),
            (0,0),
            (2,2)])
   
   a = CRSSparsity(4,5)
   for i in nza:
     a.getNZ(i[0],i[1])
     
   b = CRSSparsity(4,5)  
   for i in nzb:
     b.getNZ(i[0],i[1])
     
   w = UCharVector()
   c=a.patternUnion(b,w)
   self.assertEquals(w.size(),len(nza.union(nzb)))
   for k in range(w.size()):
     ind = (c.getRow()[k],c.col(k))
     if (ind in nza and ind in nzb):
       self.assertEquals(w[k],1 | 2)
     elif (ind in nza):
       self.assertEquals(w[k],1)
     elif (ind in nzb):
       self.assertEquals(w[k],2)
       
   c = a + b
   self.assertEquals(c.size(),len(nza.union(nzb)))
   for k in range(c.size()):
     ind = (c.getRow()[k],c.col(k))
     self.assertTrue(ind in nza or ind in nzb)