restore seqn_key usage in to_mp7_pathlines, ignore irpt in roundtrip …

…test
modflowpy · Nov 27, 2023 · b4340fb · b4340fb
1 parent 2ea4124
commit b4340fb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/autotest/test_plotutil.py b/autotest/test_plotutil.py
@@ -299,9 +299,13 @@ def test_to_prt_pathlines_roundtrip():
     mp7_pls = to_mp7_pathlines(inp_pls)
     prt_pls = to_prt_pathlines(mp7_pls)
     inp_pls.drop(
-        ["imdl", "iprp", "name", "istatus", "ireason"], axis=1, inplace=True
+        ["imdl", "iprp", "irpt", "name", "istatus", "ireason"],
+        axis=1,
+        inplace=True,
     )
     prt_pls.drop(
-        ["imdl", "iprp", "name", "istatus", "ireason"], axis=1, inplace=True
+        ["imdl", "iprp", "irpt", "name", "istatus", "ireason"],
+        axis=1,
+        inplace=True,
     )
     assert np.allclose(inp_pls, prt_pls)
diff --git a/flopy/plot/plotutil.py b/flopy/plot/plotutil.py
@@ -2691,8 +2691,7 @@ def to_mp7_pathlines(
     # build mp7 format recarray
     ret = np.core.records.fromarrays(
         [
-            data["irpt"],
-            # data[seqn_key],
+            data[seqn_key],
             data["iprp"],
             data[seqn_key],
             data["irpt"],
@@ -2924,7 +2923,7 @@ def to_prt_pathlines(
             data["timestep"],
             np.zeros(data.shape[0]),
             data["particlegroup"],
-            data["particleid"],
+            data["sequencenumber"],
             data["k"],
             data["node"],
             np.zeros(data.shape[0]),  # todo izone?