Skip to content

Commit 903a743

Browse files
authored
Fix race condition in tests
Fix race condition in tests and minor changes
1 parent 4ffe723 commit 903a743

File tree

4 files changed

+18
-12
lines changed

4 files changed

+18
-12
lines changed

.github/workflows/main.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,9 @@ jobs:
2222
uses: actions/checkout@v3
2323
- uses: jpribyl/[email protected]
2424
with:
25-
key: ${{ matrix.shmem }}-2-{hash}
25+
key: ${{ matrix.shmem }}-3-{hash}
2626
restore-keys: |
27-
${{ matrix.shmem }}-2-
27+
${{ matrix.shmem }}-3-
2828
- name: Build Docker container
2929
run: docker build -t local docker/${{ matrix.shmem }}/
3030
- name: run shmem4py test-1

test/test_buf.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ def testGetBuffer(self):
1111
self.assertTrue(isinstance(cdata, shmem.ffi.CData))
1212
self.assertEqual(csize, 4)
1313
self.assertEqual(ctype, 'double')
14-
14+
1515
def testReadOnly(self):
1616
a = np.zeros(1)
1717
a.flags.writeable = False

test/test_ptr.py

+12-6
Original file line numberDiff line numberDiff line change
@@ -23,24 +23,26 @@ def testPtrMemory(self):
2323
for t in types:
2424
sym = shmem.new_array(1, t)
2525
sym[0] = npes
26-
shmem.barrier_all()
26+
shmem.sync_all()
2727

2828
nloc = shmem.ptr(sym, nxpe)
2929
if nloc is not None:
3030
self.assertTrue(isinstance(nloc, np.ndarray))
3131
self.assertEqual(nloc.dtype, sym.dtype)
3232
self.assertEqual(nloc[0], npes)
3333
nloc[0] = nxpe
34-
shmem.barrier_all()
34+
shmem.sync_all()
3535
self.assertEqual(sym[0], mype)
3636

37+
shmem.sync_all()
38+
3739
ploc = shmem.ptr(sym, pvpe)
3840
if ploc is not None:
3941
self.assertTrue(isinstance(ploc, np.ndarray))
4042
self.assertEqual(ploc.dtype, sym.dtype)
4143
self.assertEqual(ploc[0], pvpe)
4244
ploc[0] = mype
43-
shmem.barrier_all()
45+
shmem.sync_all()
4446
self.assertEqual(sym[0], nxpe)
4547

4648
shmem.free(sym)
@@ -54,7 +56,7 @@ def testPtrArray(self):
5456
for d in (0, 1, 2, 3):
5557
for order in ('C', 'F'):
5658
sym = shmem.full((3,)*d, npes, dtype=t, order=order)
57-
shmem.barrier_all()
59+
5860
loc = shmem.ptr(sym, nxpe)
5961
if loc is not None:
6062
self.assertTrue(isinstance(loc, np.ndarray))
@@ -63,8 +65,11 @@ def testPtrArray(self):
6365
self.assertEqual(loc.strides, sym.strides)
6466
self.assertTrue(np.all(loc == npes))
6567
loc[...] = nxpe
66-
shmem.barrier_all()
68+
shmem.sync_all()
6769
self.assertTrue(np.all(sym == mype))
70+
71+
shmem.sync_all()
72+
6873
loc = shmem.ptr(sym, pvpe)
6974
if loc is not None:
7075
self.assertTrue(isinstance(loc, np.ndarray))
@@ -73,8 +78,9 @@ def testPtrArray(self):
7378
self.assertEqual(loc.strides, sym.strides)
7479
self.assertTrue(np.all(loc == pvpe))
7580
loc[...] = mype
76-
shmem.barrier_all()
81+
shmem.sync_all()
7782
self.assertTrue(np.all(sym == nxpe))
83+
7884
shmem.free(sym)
7985

8086

test/test_query.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -15,17 +15,17 @@ def testQueryThread(self):
1515
)
1616
)
1717

18-
def testPEAccesible(self):
18+
def testPEAccessible(self):
1919
flag = shmem.pe_accessible(shmem.my_pe())
2020
self.assertTrue(flag)
2121

22-
def testAddrAccesibleCData(self):
22+
def testAddrAccessibleCData(self):
2323
addr = shmem.new_array(1, 'i')
2424
flag = shmem.addr_accessible(addr, shmem.my_pe())
2525
shmem.free(addr)
2626
self.assertTrue(flag)
2727

28-
def testAddrAccesibleNumPy(self):
28+
def testAddrAccessibleNumPy(self):
2929
addr = shmem.empty(1, dtype='i')
3030
flag = shmem.addr_accessible(addr, shmem.my_pe())
3131
shmem.free(addr)

0 commit comments

Comments
 (0)