petsc4py/test/test_dmshell.py

1 from petsc4py import PETSc
7     COMM = PETSc.COMM_WORLD
10         self.dm = PETSc.DMShell().create(comm=self.COMM)
15         PETSc.garbage_cleanup()
18         vec = PETSc.Vec().create(comm=self.COMM)
28             v = PETSc.Vec().create(comm=dm.comm)
39         vec = PETSc.Vec().create(comm=PETSc.COMM_SELF)
47         self.assertEqual(lvec.comm, PETSc.COMM_SELF)
51             v = PETSc.Vec().create(comm=PETSc.COMM_SELF)
61         self.assertEqual(lvec.comm, PETSc.COMM_SELF)
64         mat = PETSc.Mat().create(comm=self.COMM)
74             mat = PETSc.Mat().create(comm=self.COMM)
85             return None, [PETSc.IS().createStride(1) for _ in range(2)], None
121                     self.assertEqual(type(dm), PETSc.DM)
127             if mode == PETSc.InsertMode.INSERT_VALUES:
129             elif mode == PETSc.InsertMode.ADD_VALUES:
135         vec = PETSc.Vec().create(comm=self.COMM)
139         ovec = PETSc.Vec().create(comm=PETSc.COMM_SELF)
143         self.dm.globalToLocal(vec, ovec, addv=PETSc.InsertMode.INSERT_VALUES)
145         self.dm.globalToLocal(vec, ovec, addv=PETSc.InsertMode.ADD_VALUES)
150             if mode == PETSc.InsertMode.INSERT_VALUES:
152             elif mode == PETSc.InsertMode.ADD_VALUES:
158         vec = PETSc.Vec().create(comm=PETSc.COMM_SELF)
162         ovec = PETSc.Vec().create(comm=self.COMM)
166         self.dm.localToGlobal(vec, ovec, addv=PETSc.InsertMode.INSERT_VALUES)
168         self.dm.localToGlobal(vec, ovec, addv=PETSc.InsertMode.ADD_VALUES)
173             if mode == PETSc.InsertMode.INSERT_VALUES:
175             elif mode == PETSc.InsertMode.ADD_VALUES:
181         vec = PETSc.Vec().create(comm=PETSc.COMM_SELF)
187         self.dm.localToLocal(vec, ovec, addv=PETSc.InsertMode.INSERT_VALUES)
189         self.dm.localToLocal(vec, ovec, addv=PETSc.InsertMode.ADD_VALUES)
193         vec = PETSc.Vec().create()
196         sct, ovec = PETSc.Scatter.toAll(vec)
198         self.dm.globalToLocal(vec, ovec, addv=PETSc.InsertMode.INSERT_VALUES)
201         vec = PETSc.Vec().create()
204         sct, ovec = PETSc.Scatter.toAll(vec)
206         self.dm.localToGlobal(vec, ovec, addv=PETSc.InsertMode.INSERT_VALUES)
209         vec = PETSc.Vec().create()
212         sct, ovec = PETSc.Scatter.toAll(vec)
214         self.dm.localToLocal(vec, ovec, addv=PETSc.InsertMode.INSERT_VALUES)
217         cdm = PETSc.DMShell().create(comm=self.COMM)
233         mat = PETSc.Mat().create()
236         vec = PETSc.Vec().create()
249         mat = PETSc.Mat().create()