File tree Expand file tree Collapse file tree 1 file changed +11
-8
lines changed Expand file tree Collapse file tree 1 file changed +11
-8
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments