Update tests/test_basic.py

Co-authored-by: Peter Hill <[email protected]>
PlasmaFAIR · Aug 2, 2024 · 4967ebc · 4967ebc
1 parent e2f7eba
commit 4967ebc
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_basic.py b/tests/test_basic.py
@@ -16,9 +16,9 @@ def test_basic():
 
 
 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
+    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"