Merge pull request #61 from nschloe/fix-relative-to

Fix relative to

perfplot/__about__.py

@@ -2,5 +2,5 @@
 __author_email__ = "nico.schloemer@gmail.com"
 __copyright__ = "Copyright (c) 2017-2019, {} <{}>".format(__author__, __author_email__)
 __license__ = "License :: OSI Approved :: MIT License"
-__version__ = "0.6.8"
+__version__ = "0.6.9"
 __status__ = "Development Status :: 5 - Production/Stable"

perfplot/main.py

@@ -81,7 +81,7 @@ def plot(
             # correspond to the order of the lines.
             order = numpy.argsort(self.timings[:, -1])[::-1]
             if relative_to is not None:
-                relative_to = order[relative_to]
+                relative_to = numpy.where(order == relative_to)[0][0]
             self.timings = self.timings[order]
             self.labels = [self.labels[i] for i in order]
             self.colors = [self.colors[i] for i in order]
@@ -159,6 +159,9 @@ def bench(
                 relative_to = kernels[0](data)
             for k, kernel in enumerate(tqdm(kernels, leave=(i == len(n_range) - 1))):
                 if equality_check:
+                    assert kernel(data) is not None, "{} returned None".format(
+                        labels[k]
+                    )
                     assert equality_check(
                         relative_to, kernel(data)
                     ), "Equality check failure. ({}, {})".format(labels[0], labels[k])