more information from cuda test failures

osrf · Jan 24, 2025 · 2e34e83 · 2e34e83
1 parent 6c6bc2d
commit 2e34e83
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_nvidia.py b/test/test_nvidia.py
@@ -300,8 +300,8 @@ def test_cuda_install(self):
         active_extensions = [e() for e in plugins.values() if e.get_name() in desired_plugins]
         for tag in self.dockerfile_tags:
             dig = DockerImageGenerator(active_extensions, {}, tag)
-            self.assertEqual(dig.build(), 0)
-            self.assertEqual(dig.run(), 0)
+            self.assertEqual(dig.build(), 0, msg=f'Failed to dig.build() for {tag}')
+            self.assertEqual(dig.run(), 0, msg=f'Failed to dig.run() for {tag}')
             dig.clear_image()
 
     def test_cuda_env_subs(self):