Skip to content

Commit 2579b3e

Browse files
committed
Fix test_darray h5py issue
1 parent d492eac commit 2579b3e

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

tests/test_darray.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -84,14 +84,17 @@ def test_3Darray():
8484
assert a0.rank == 0
8585
aa = a.v
8686
assert isinstance(aa, np.ndarray)
87-
k = a.get((0,)*rank+(0, 0, slice(None)))
88-
if comm.Get_rank() == 0:
89-
assert len(k) == N[2]
90-
assert np.sum(k) == N[2]
91-
k = a.get((0,)*rank+(slice(None), 0, 0))
92-
if comm.Get_rank() == 0:
93-
assert len(k) == N[0]
94-
assert np.sum(k) == N[0]
87+
try:
88+
k = a.get((0,)*rank+(0, 0, slice(None)))
89+
if comm.Get_rank() == 0:
90+
assert len(k) == N[2]
91+
assert np.sum(k) == N[2]
92+
k = a.get((0,)*rank+(slice(None), 0, 0))
93+
if comm.Get_rank() == 0:
94+
assert len(k) == N[0]
95+
assert np.sum(k) == N[0]
96+
except ModuleNotFoundError:
97+
pass
9598
_ = a.local_slice()
9699
newaxis = (a.alignment+1)%3
97100
_ = a.get_pencil_and_transfer(newaxis)

0 commit comments

Comments
 (0)