fix a few unittests

xadupre · xadupre · commit 2ba17ff3d0ef · 2023-09-18T17:20:43.000+02:00
diff --git a/_doc/conf.py b/_doc/conf.py
@@ -2,7 +2,7 @@
 import sys
 from sphinx_runpython.github_link import make_linkcode_resolve
 from sphinx_runpython.conf_helper import has_dvipng, has_dvisvgm
-from mlinsights import __version__, has_cuda
+from mlinsights import __version__
 
 extensions = [
     "sphinx.ext.autodoc",
@@ -50,7 +50,7 @@
 
 
 def setup(app):
-    app.add_config_value("HAS_CUDA", "1" if has_cuda() else "0", "env")
+    app.add_config_value("HAS_CUDA", "0")
 
 
 html_theme = "furo"
diff --git a/_doc/examples/plot_search_images_torch.py b/_doc/examples/plot_search_images_torch.py
@@ -70,6 +70,8 @@
 
 url = "https://github.com/sdpython/mlinsights/raw/ref/_doc/examples/data/dog-cat-pixabay.zip"
 files = unzip_files(url, where_to="simages/category")
+if len(files) == 0:
+    raise FileNotFoundError(f"No images where unzipped from {url!r}.")
 len(files), files[0]
 
 ##########################################
diff --git a/_unittests/ut_search_rank/test_LONG_search_images_torch.py b/_unittests/ut_search_rank/test_LONG_search_images_torch.py
@@ -28,21 +28,20 @@ def test_search_predictions_torch(self):
         model = tmodels.squeezenet1_1(pretrained=True)
 
         # images
+        this = os.path.dirname(__file__)
         with tempfile.TemporaryDirectory() as temp:
-            dest = os.path.join(temp, "simages")
-            os.mkdir(dest)
-            zipname = os.path.join(
-                temp,
-                "..",
+            sub = os.path.join(temp, "simages")
+            os.mkdir(sub)
+            zipimg = os.path.join(
+                this,
                 "..",
                 "..",
                 "_doc",
-                "notebooks",
-                "explore",
+                "examples",
                 "data",
                 "dog-cat-pixabay.zip",
             )
-            files = unzip_files(zipname, where_to=dest)
+            files = unzip_files(zipimg, where_to=sub)
             self.assertTrue(len(files) > 0)
 
             # sequence of images
diff --git a/mlinsights/ext_test_case.py b/mlinsights/ext_test_case.py
@@ -307,6 +307,10 @@ def assertEqualFloat(self, a, b, precision=1e-5):
             except AssertionError:
                 raise AssertionError(f"{a} != {b} (p={precision})")
 
+    def assertEndsWith(self, suffix: str, text: str):
+        if not text.endswith(suffix):
+            raise AssertionError(f"Unable to find {suffix!r} in {text!r}.")
+
     @classmethod
     def tearDownClass(cls):
         for name, line, w in cls._warns:
diff --git a/mlinsights/helpers/pipeline.py b/mlinsights/helpers/pipeline.py
@@ -210,5 +210,5 @@ def decision_function(self, X, *args, **kwargs):
                 setattr(model, k, MethodType(new_methods[k], model))
             except AttributeError:
                 warnings.warn(
-                    f"Unable to overwrite method {k!r} for class " f"{type(model)!r}."
+                    f"Unable to overwrite method {k!r} for class {type(model)!r}."
                 )

Original file line number	Diff line number	Diff line change
`@@ -210,5 +210,5 @@ def decision_function(self, X, args, *kwargs):`
`210`	`210`	`setattr(model, k, MethodType(new_methods[k], model))`
`211`	`211`	`except AttributeError:`
`212`	`212`	`warnings.warn(`
`213`		`- f"Unable to overwrite method {k!r} for class " f"{type(model)!r}."`
	`213`	`+ f"Unable to overwrite method {k!r} for class {type(model)!r}."`
`214`	`214`	`)`