Lines Matching refs:vec

18         vec = PETSc.Vec().create(comm=self.COMM)
19         vec.setSizes((10, None))
20         vec.setUp()
21         self.dm.setGlobalVector(vec)
23         self.assertEqual(vec.getSizes(), gvec.getSizes())
24         self.assertEqual(vec.comm, gvec.comm)
39         vec = PETSc.Vec().create(comm=PETSc.COMM_SELF)
40         vec.setSizes((1 + 10 * self.COMM.rank, None))
41         vec.setUp()
42         self.dm.setLocalVector(vec)
44         self.assertEqual(vec.getSizes(), lvec.getSizes())
135         vec = PETSc.Vec().create(comm=self.COMM)
136         vec.setSizes((10, None))
137         vec.setUp()
138         vec[...] = self.dm.comm.rank + 1
143         self.dm.globalToLocal(vec, ovec, addv=PETSc.InsertMode.INSERT_VALUES)
144         self.assertTrue(np.allclose(vec.getArray(), ovec.getArray()))
145         self.dm.globalToLocal(vec, ovec, addv=PETSc.InsertMode.ADD_VALUES)
146         self.assertTrue(np.allclose(2 * vec.getArray(), ovec.getArray()))
158         vec = PETSc.Vec().create(comm=PETSc.COMM_SELF)
159         vec.setSizes((10, None))
160         vec.setUp()
161         vec[...] = self.dm.comm.rank + 1
166         self.dm.localToGlobal(vec, ovec, addv=PETSc.InsertMode.INSERT_VALUES)
167         self.assertTrue(np.allclose(vec.getArray(), ovec.getArray()))
168         self.dm.localToGlobal(vec, ovec, addv=PETSc.InsertMode.ADD_VALUES)
169         self.assertTrue(np.allclose(2 * vec.getArray(), ovec.getArray()))
181         vec = PETSc.Vec().create(comm=PETSc.COMM_SELF)
182         vec.setSizes((10, None))
183         vec.setUp()
184         vec[...] = self.dm.comm.rank + 1
185         ovec = vec.duplicate()
187         self.dm.localToLocal(vec, ovec, addv=PETSc.InsertMode.INSERT_VALUES)
188         self.assertTrue(np.allclose(vec.getArray(), ovec.getArray()))
189         self.dm.localToLocal(vec, ovec, addv=PETSc.InsertMode.ADD_VALUES)
190         self.assertTrue(np.allclose(2 * vec.getArray(), ovec.getArray()))
193         vec = PETSc.Vec().create()
194         vec.setSizes((10, None))
195         vec.setUp()
196         sct, ovec = PETSc.Scatter.toAll(vec)
198         self.dm.globalToLocal(vec, ovec, addv=PETSc.InsertMode.INSERT_VALUES)
201         vec = PETSc.Vec().create()
202         vec.setSizes((10, None))
203         vec.setUp()
204         sct, ovec = PETSc.Scatter.toAll(vec)
206         self.dm.localToGlobal(vec, ovec, addv=PETSc.InsertMode.INSERT_VALUES)
209         vec = PETSc.Vec().create()
210         vec.setSizes((10, None))
211         vec.setUp()
212         sct, ovec = PETSc.Scatter.toAll(vec)
214         self.dm.localToLocal(vec, ovec, addv=PETSc.InsertMode.INSERT_VALUES)
236         vec = PETSc.Vec().create()
237         vec.setSizes((10, None))
238         vec.setUp()
241             return mat, vec
246         self.assertEqual(v, vec)