Fixing coverage

Author: mikaem

tests/test_darray.py

@@ -33,14 +33,14 @@ def test_2Darray():
                 assert a0.rank == rank-1
             aa = a.v
             assert isinstance(aa, np.ndarray)
-            #k = a.get((0,)*rank+(0, slice(None)))
-            #if comm.Get_rank() == 0:
-            #    assert len(k) == N[1]
-            #    assert np.sum(k) == N[1]
-            #k = a.get((0,)*rank+(slice(None), 0))
-            #if comm.Get_rank() == 0:
-            #    assert len(k) == N[0]
-            #    assert np.sum(k) == N[0]
+            k = a.get((0,)*rank+(0, slice(None)))
+            if comm.Get_rank() == 0:
+                assert len(k) == N[1]
+                assert np.sum(k) == N[1]
+            k = a.get((0,)*rank+(slice(None), 0))
+            if comm.Get_rank() == 0:
+                assert len(k) == N[0]
+                assert np.sum(k) == N[0]
             ls = a.local_slice()
             newaxis = (a.alignment+1)%2
             p0, t = a.get_pencil_and_transfer(newaxis)
@@ -81,14 +81,14 @@ def test_3Darray():
                 assert a0.rank == 0
             aa = a.v
             assert isinstance(aa, np.ndarray)
-            #k = a.get((0,)*rank+(0, 0, slice(None)))
-            #if comm.Get_rank() == 0:
-            #    assert len(k) == N[2]
-            #    assert np.sum(k) == N[2]
-            #k = a.get((0,)*rank+(slice(None), 0, 0))
-            #if comm.Get_rank() == 0:
-            #    assert len(k) == N[0]
-            #    assert np.sum(k) == N[0]
+            k = a.get((0,)*rank+(0, 0, slice(None)))
+            if comm.Get_rank() == 0:
+                assert len(k) == N[2]
+                assert np.sum(k) == N[2]
+            k = a.get((0,)*rank+(slice(None), 0, 0))
+            if comm.Get_rank() == 0:
+                assert len(k) == N[0]
+                assert np.sum(k) == N[0]
             ls = a.local_slice()
             newaxis = (a.alignment+1)%3
             p0, t = a.get_pencil_and_transfer(newaxis)

tests/test_io.py

@@ -56,6 +56,9 @@ def test_2D(backend, forward_output):
             u0.read(filename, 'u', 2)
             u0.read(read, 'u', 2)
             assert np.allclose(u0, u)
+            if backend == 'netcdf4': # Test opening file in mode 'a' when not existing
+                remove_if_exists('nctesta.nc')
+                ncfile = NCFile('nctesta.nc', domain=domain, mode='a')
     T.destroy()
 
 def test_3D(backend, forward_output):