add logical "==" to filter_modpath_by_travel_time() (#1736)

flopy/plot/crosssection.py

@@ -1068,7 +1068,7 @@ def plot_pathline(
             less than or equal to the passed time are plotted. If a
             string is passed a variety logical constraints can be added
             in front of a time value to select pathlines for a select
-            period of time. Valid logical constraints are <=, <, >=, and
+            period of time. Valid logical constraints are <=, <, ==, >=, and
             >. For example, to select all pathlines less than 10000 days
             travel_time='< 10000' would be passed to plot_pathline.
             (default is None)
@@ -1184,7 +1184,7 @@ def plot_timeseries(
             less than or equal to the passed time are plotted. If a
             string is passed a variety logical constraints can be added
             in front of a time value to select pathlines for a select
-            period of time. Valid logical constraints are <=, <, >=, and
+            period of time. Valid logical constraints are <=, <, ==, >=, and
             >. For example, to select all pathlines less than 10000 days
             travel_time='< 10000' would be passed to plot_pathline.
             (default is None)

flopy/plot/map.py

@@ -695,7 +695,7 @@ def plot_pathline(self, pl, travel_time=None, **kwargs):
             less than or equal to the passed time are plotted. If a
             string is passed a variety logical constraints can be added
             in front of a time value to select pathlines for a select
-            period of time. Valid logical constraints are <=, <, >=, and
+            period of time. Valid logical constraints are <=, <, ==, >=, and
             >. For example, to select all pathlines less than 10000 days
             travel_time='< 10000' would be passed to plot_pathline.
             (default is None)
@@ -809,7 +809,7 @@ def plot_timeseries(self, ts, travel_time=None, **kwargs):
             less than or equal to the passed time are plotted. If a
             string is passed a variety logical constraints can be added
             in front of a time value to select pathlines for a select
-            period of time. Valid logical constraints are <=, <, >=, and
+            period of time. Valid logical constraints are <=, <, ==, >=, and
             >. For example, to select all pathlines less than 10000 days
             travel_time='< 10000' would be passed to plot_pathline.
             (default is None)

flopy/plot/plotutil.py

@@ -2571,10 +2571,19 @@ def advanced_package_bc_helper(pkg, modelgrid, kper):
 
 def filter_modpath_by_travel_time(recarray, travel_time):
     """
+    Helper method for filtering particles by travel time. Used in modpath
+    plotting routines
 
-    :param recarray:
-    :param travel_time:
-    :return:
+    Parameters
+    ----------
+    recarray : np.recarray
+        recarray of modpath particle information
+    travel_time : str, float
+        travel time logical argument to filter modpath output
+
+    Returns
+    -------
+        np.recarray
     """
     if travel_time is None:
         tp = recarray.copy()
@@ -2583,6 +2592,7 @@ def filter_modpath_by_travel_time(recarray, travel_time):
             funcs = {
                 "<=": lambda a, b: a["time"] <= b,
                 ">=": lambda a, b: a["time"] >= b,
+                "==": lambda a, b: a["time"] == b,
                 "<": lambda a, b: a["time"] < b,
                 ">": lambda a, b: a["time"] > b,
             }
@@ -2600,7 +2610,7 @@ def filter_modpath_by_travel_time(recarray, travel_time):
                     raise Exception(
                         "flopy.map.plot_pathline travel_time variable cannot "
                         "be parsed. Acceptable logical variables are , "
-                        "<=, <, >=, and >. "
+                        "<=, <, ==, >=, and >. "
                         "You passed {}".format(travel_time)
                     )
         else: