corrected helper behaviour if single station is given

mlair/helpers/helpers.py

@@ -32,6 +32,11 @@ def dict_to_xarray(d: Dict, coordinate_name: str) -> xr.DataArray:
 
     :return: combined xarray
     """
+    if len(d.keys()) == 1:
+        k = list(d.keys())
+        xarray: xr.DataArray = d[k[0]]
+        return xarray.expand_dims(dim={coordinate_name: k}, axis=0)
+    else:
         xarray = None
         for k, v in d.items():
             if xarray is None:

test/test_helpers/test_helpers.py

@@ -124,14 +124,22 @@ class TestPytestRegex:
 class TestDictToXarray:
 
     def test_dict_to_xarray(self):
-        array1 = xr.DataArray(np.random.randn(2, 3), dims=('x', 'y'), coords={'x': [10, 20]})
-        array2 = xr.DataArray(np.random.randn(2, 3), dims=('x', 'y'), coords={'x': [10, 20]})
+        array1 = xr.DataArray(np.random.randn(2, 3), dims=('x', 'y'), coords={'x': [10, 20], 'y': [0, 10, 20]})
+        array2 = xr.DataArray(np.random.randn(2, 3), dims=('x', 'y'), coords={'x': [10, 20], 'y': [0, 10, 20]})
         d = {"number1": array1, "number2": array2}
         res = dict_to_xarray(d, "merge_dim")
         assert type(res) == xr.DataArray
-        assert sorted(list(res.coords)) == ["merge_dim", "x"]
+        assert sorted(list(res.coords)) == ["merge_dim", "x", "y"]
         assert res.shape == (2, 2, 3)
 
+    def test_dict_to_xarray_single_entry(self):
+        array1 = xr.DataArray(np.random.randn(2, 3), dims=('x', 'y'), coords={'x': [10, 20], 'y': [0, 10, 20]})
+        d = {"number1": array1}
+        res = dict_to_xarray(d, "merge_dim")
+        assert type(res) == xr.DataArray
+        assert sorted(list(res.coords)) == ["merge_dim", "x", "y"]
+        assert res.shape == (1, 2, 3)
+
 
 class TestFloatRound: