Merge pull request #218 from dstansby/xarray-smoke

Add some xarray smoke tests

tests/conftest.py

@@ -0,0 +1,11 @@
+import pathlib
+
+import pytest
+
+
+@pytest.fixture(params=["psp_fld_l2_mag_rtn_1min_20200104_v02.cdf", "de2_ion2s_rpa_19830213_v01.cdf"])
+def cdf_path(request):
+    """
+    Returns a series of CDF file paths which can be used for smoke testing.
+    """
+    return (pathlib.Path(__file__) / ".." / "testfiles" / request.param).resolve()

tests/test_cdfread.py

@@ -1,14 +1,10 @@
-import pathlib
-
 import pytest
 
 from cdflib import CDF
 
 
-@pytest.mark.parametrize("fname", ["psp_fld_l2_mag_rtn_1min_20200104_v02.cdf", "de2_ion2s_rpa_19830213_v01.cdf"])
-def test_read(fname):
-    fname = (pathlib.Path(__file__) / ".." / "testfiles" / fname).resolve()
-    cdf = CDF(fname)
+def test_read(cdf_path):
+    cdf = CDF(cdf_path)
 
     info = cdf.cdf_info()
 
@@ -18,7 +14,7 @@ def test_read(fname):
         cdf.varget(var)
 
     # Smoke test context manager
-    with CDF(fname) as cdf:
+    with CDF(cdf_path) as cdf:
         cdf.cdf_info()
 
     # Smoke test global attributes

tests/test_xarray_reader_writer.py

@@ -347,3 +347,8 @@ def test_build_from_scratch():
     ds = xr.Dataset(data_vars={"data": data, "epoch": epoch, "direction": direction}, attrs=global_attributes)
     xarray_to_cdf(ds, "hello.cdf")
     os.remove("hello.cdf")
+
+
+def test_smoke(cdf_path, tmp_path):
+    a = cdf_to_xarray(cdf_path, to_unixtime=True, fillval_to_nan=True)
+    xarray_to_cdf(a, tmp_path / cdf_path.name, from_unixtime=True)