1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
from mpi4py import MPI
import mpiunittest as unittest
class GReqCtx(object):
source = 3
tag = 7
completed = False
cancel_called = False
free_called = False
def query(self, status):
status.Set_source(self.source)
status.Set_tag(self.tag)
def free(self):
self.free_called = True
def cancel(self, completed):
self.cancel_called = True
if completed is not self.completed:
raise MPI.Exception(MPI.ERR_PENDING)
class TestGrequest(unittest.TestCase):
def testAll(self):
ctx = GReqCtx()
greq = MPI.Grequest.Start(ctx.query, ctx.free, ctx.cancel)
self.assertFalse(greq.Test())
self.assertFalse(ctx.free_called)
greq.Cancel()
self.assertTrue(ctx.cancel_called)
ctx.cancel_called = False
greq.Complete()
ctx.completed = True
greq.Cancel()
self.assertTrue(ctx.cancel_called)
status = MPI.Status()
self.assertTrue(greq.Test(status))
self.assertEqual(status.Get_source(), ctx.source)
self.assertEqual(status.Get_tag(), ctx.tag)
self.assertEqual(status.Get_error(), MPI.SUCCESS)
greq.Wait()
self.assertTrue(ctx.free_called)
def testAll1(self):
ctx = GReqCtx()
greq = MPI.Grequest.Start(ctx.query, None, None)
self.assertFalse(greq.Test())
greq.Cancel()
greq.Complete()
status = MPI.Status()
self.assertTrue(greq.Test(status))
self.assertEqual(status.Get_source(), ctx.source)
self.assertEqual(status.Get_tag(), ctx.tag)
self.assertEqual(status.Get_error(), MPI.SUCCESS)
self.assertFalse(status.Is_cancelled())
greq.Wait()
def testAll2(self):
greq = MPI.Grequest.Start(None, None, None)
self.assertFalse(greq.Test())
greq.Cancel()
greq.Complete()
status = MPI.Status()
self.assertTrue(greq.Test(status))
self.assertEqual(status.Get_source(), MPI.ANY_SOURCE)
self.assertEqual(status.Get_tag(), MPI.ANY_TAG)
self.assertEqual(status.Get_error(), MPI.SUCCESS)
self.assertFalse(status.Is_cancelled())
greq.Wait()
if MPI.Get_version() < (2, 0):
del GReqCtx
del TestGrequest
if __name__ == '__main__':
unittest.main()
|
