From 4967ebcaa913a44dcac30700cfd1845e2021b99f Mon Sep 17 00:00:00 2001 From: Joel Adams <57064123+JoelLucaAdams@users.noreply.github.com> Date: Fri, 2 Aug 2024 13:58:39 +0100 Subject: [PATCH] Update tests/test_basic.py Co-authored-by: Peter Hill --- tests/test_basic.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/test_basic.py b/tests/test_basic.py index 9fe4256..d3a103a 100644 --- 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"