Merge pull request #36 from wx4stg/datasetonefile

lma_read.dataset single file support

Author: deeplycloudy

pyxlma/lmalib/io/read.py

@@ -97,6 +97,8 @@ def dataset(filenames, sort_time=True):
     """ Create an xarray dataset of the type returned by
         pyxlma.lmalib.io.cf_netcdf.new_dataset for each filename in filenames
     """
+    if type(filenames) == str:
+        filenames = [filenames]
     lma_data = []
     starttime = None
     next_event_id = 0

tests/test_io.py

@@ -4,6 +4,7 @@
 from datetime import datetime as dt
 import xarray as xr
 from pyxlma.lmalib.io import read as lma_read
+from tests.test_flash import compare_dataarrays
 
 
 def test_read_mf_dataset():
@@ -13,3 +14,11 @@ def test_read_mf_dataset():
     dataset, start_date = lma_read.dataset(files_to_read)
     assert start_date == dt(2023, 12, 24, 0, 57, 1)
     assert dataset == xr.open_dataset('tests/truth/lma_netcdf/lma.nc')
+
+
+def test_read_onefile_dataset():
+    dataset, start_date = lma_read.dataset('examples/data/WTLMA_231224_005701_0001.dat.gz')
+    assert start_date == dt(2023, 12, 24, 0, 57, 1)
+    truth = xr.open_dataset('tests/truth/lma_netcdf/small_lma.nc')
+    for var in truth.data_vars:
+        compare_dataarrays(dataset, truth, var)