Updated tests

PlasmaFAIR · Aug 2, 2024 · b7feeac · b7feeac
1 parent 324858d
commit b7feeac
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/tests/example_files/0011.sdf b/tests/example_files/0011.sdf
diff --git a/tests/test_basic.py b/tests/test_basic.py
@@ -10,6 +10,14 @@ def test_basic():
     with xr.open_dataset(EXAMPLE_FILES_DIR / "0000.sdf") as df:
         ex_field = "Electric Field/Ex"
         assert ex_field in df
-        x_coord = "X_x_px_deltaf/electron_beam"
+        x_coord = "X_Grid_mid"
         assert x_coord in df[ex_field].coords
         assert df[x_coord].attrs["long_name"] == "X"
+
+def test_coords():
+    with xr.open_dataset(EXAMPLE_FILES_DIR / "0011.sdf") as df:
+        px_photon = "dist_fn/px_py/Photon"
+        assert px_photon in df
+        x_coord = "Px_px_py/Photon"
+        assert x_coord in df[px_photon].coords
+        assert df[x_coord].attrs["long_name"] == "Px"