Merge pull request #142 from nschloe/string

string
nschloe · Mar 3, 2022 · 5c43143 · 5c43143
2 parents 2c55564 + a1b41c7
commit 5c43143
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/src/perfplot/__about__.py b/src/perfplot/__about__.py
@@ -1 +1 @@
-__version__ = "0.10.1"
+__version__ = "0.10.2"
diff --git a/src/perfplot/_main.py b/src/perfplot/_main.py
@@ -33,6 +33,13 @@
 }
 
 
+def default_equality_check(a, b):
+    if isinstance(a, str):
+        return a == b
+
+    return np.allclose(a, b)
+
+
 def _auto_time_unit(time_s: float) -> str:
     """Automatically obtains a readable unit at which to plot :py:attr:`timings` of the
     benchmarking process. This is accomplished by converting the minimum measured
@@ -351,7 +358,7 @@ def live(
     xlabel: str | None = None,
     target_time_per_measurement: float = 1.0,
     max_time: float | None = None,
-    equality_check: Callable | None = np.allclose,
+    equality_check: Callable | None = default_equality_check,
     show_progress: bool = True,
     logx: Literal["auto"] | bool = "auto",
     logy: Literal["auto"] | bool = "auto",
@@ -472,7 +479,7 @@ def bench(
     title: str | None = None,
     target_time_per_measurement: float = 1.0,
     max_time: float | None = None,
-    equality_check: Callable | None = np.allclose,
+    equality_check: Callable | None = default_equality_check,
     show_progress: bool = True,
 ):
     if labels is None: