Updated tests

PlasmaFAIR · Aug 2, 2024 · d65c66d · d65c66d
1 parent 324858d
commit d65c66d
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/tests/test_basic.py b/tests/test_basic.py
@@ -10,6 +10,15 @@ 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 / "0010.sdf") as df:
+        px_electron = "dist_fn/x_px/electron"
+        assert px_electron in df
+        x_coord = "Px_x_px/electron"
+        assert x_coord in df[px_electron].coords
+        assert df[x_coord].attrs["long_name"] == "Px"