img subset and writing test for L4

src/smos/smos_l4/interface.py

@@ -303,7 +303,10 @@ def write(self, image, **kwargs):
                              ('subset_software', f"{dist_name} | {subdist_name} | {__version__}")])
             glob_attrs = self.glob_attrs
             for k in ['ease_global', 'history', 'creation_time', 'NCO']:
-                glob_attrs.pop(k)
+                try:
+                    glob_attrs.pop(k)
+                except KeyError:
+                    continue
             glob_attrs.update(this_global_attrs)
             ds.setncatts(glob_attrs)
 

tests/test_L4img_subset_and_write.py

@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+# The MIT License (MIT)
+#
+# Copyright (c) 2021,TU Wien
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+"""
+Tests for storing subset images.
+"""
+
+from smos.smos_l4 import SMOSL4Ds
+import os
+import numpy as np
+import numpy.testing as nptest
+from datetime import datetime
+from tempfile import mkdtemp
+from smos.grid import EASE25CellGrid
+import pytest
+
+
+@pytest.mark.parametrize("stackfile,files", [(None, 1), ("stack.nc", 1)])
+def test_SMOS_L4OPER_Ds_subset(stackfile, files):
+    # test reading and storing a spatial subset (as stack/single img)
+    dsname = os.path.join(os.path.dirname(__file__),
+                          'smos-test-data', 'L4_SMOS_RZSM', 'OPER')
+    subgrid = EASE25CellGrid(bbox=(-11., 34., 43., 71.))
+    params = ['RZSM', 'Quality']
+
+    ds = SMOSL4Ds(dsname, parameters=params, grid=subgrid, oper=True)
+    image = ds.read(datetime(2020, 1, 31))
+
+    # europe subset
+    assert image.data['RZSM'].shape == (113, 208)
+    assert list(image.data.keys()) == params
+
+    assert np.nanmax(np.nanmax(image.data['Quality'])) == 1
+    assert np.nanmax(np.nanmin(image.data['Quality'])) == 0
+
+    # index in global file [508, 772]
+    nptest.assert_almost_equal(image.data['RZSM'][60, 120], 0.18237862, decimal=4)
+    nptest.assert_almost_equal(image.lon[60, 120], 20.36023, decimal=4)
+    nptest.assert_almost_equal(image.lat[60, 120], 47.682177, decimal=4)
+
+    nptest.assert_almost_equal(image.lon[25, 130], 22.95389, 4)
+    nptest.assert_almost_equal(image.lat[25, 130], 59.1181825, 4)
+    assert np.isnan(image.data['RZSM'][25, 130])
+
+    outdir = mkdtemp()
+    os.makedirs(outdir, exist_ok=True)
+    ds.write_multiple(root_path=outdir, start_date=datetime(2020, 1, 31),
+                      end_date=datetime(2020, 2, 1), stackfile=stackfile)
+
+    if stackfile is None:
+        assert os.path.isdir(os.path.join(outdir, '2020'))
+        assert len(os.listdir(os.path.join(outdir, '2020'))) == 1  # Feb 1st has no data
+    else:
+        ds.write_multiple(root_path=outdir, start_date=datetime(2020, 1, 31),
+                          end_date=datetime(2020, 2, 1), stackfile=stackfile)
+        assert len(os.listdir(outdir)) == 1  # 1 stack
+
+    print("Bazinga!")
+
+if __name__ == '__main__':
+    test_SMOS_L4OPER_Ds_subset(None, 1)
+    test_SMOS_L4OPER_Ds_subset("stack.nc", 1)