aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/argaze.test/OpenCVCuda.py10
1 files changed, 6 insertions, 4 deletions
diff --git a/src/argaze.test/OpenCVCuda.py b/src/argaze.test/OpenCVCuda.py
index fd28abf..3b5b657 100644
--- a/src/argaze.test/OpenCVCuda.py
+++ b/src/argaze.test/OpenCVCuda.py
@@ -9,10 +9,7 @@ import os
class cuda_test(unittest.TestCase):
"""Test Cuda-accelerated OpenCV functions class."""
- def test_setup(self):
-
- self.assertGreater(cv.cuda.getCudaEnabledDeviceCount(), 0)
-
+ @unittest.skipIf(cv.cuda.getCudaEnabledDeviceCount() == 0, "No cuda device found")
def test_cuda_upload_download(self):
npMat = (np.random.random((128, 128, 3)) * 255).astype(np.uint8)
@@ -21,6 +18,7 @@ class cuda_test(unittest.TestCase):
self.assertTrue(np.allclose(cuMat.download(), npMat))
+ @unittest.skipIf(cv.cuda.getCudaEnabledDeviceCount() == 0, "No cuda device found")
def test_cuda_upload_download_stream(self):
stream = cv.cuda_Stream()
@@ -32,6 +30,7 @@ class cuda_test(unittest.TestCase):
self.assertTrue(np.allclose(npMat2, npMat))
+ @unittest.skipIf(cv.cuda.getCudaEnabledDeviceCount() == 0, "No cuda device found")
def test_cuda_interop(self):
npMat = (np.random.random((128, 128, 3)) * 255).astype(np.uint8)
@@ -46,6 +45,7 @@ class cuda_test(unittest.TestCase):
asyncstream = cv.cuda_Stream(1) # cudaStreamNonBlocking
self.assertTrue(asyncstream.cudaPtr() != 0)
+ @unittest.skipIf(cv.cuda.getCudaEnabledDeviceCount() == 0, "No cuda device found")
def test_cuda_buffer_pool(self):
cv.cuda.setBufferPoolUsage(True)
@@ -58,6 +58,7 @@ class cuda_test(unittest.TestCase):
self.assertEqual(cuMat.size(), (1024, 1024))
self.assertEqual(cuMat.type(), cv.CV_8UC3)
+ @unittest.skipIf(cv.cuda.getCudaEnabledDeviceCount() == 0, "No cuda device found")
def test_cuda_release(self):
npMat = (np.random.random((128, 128, 3)) * 255).astype(np.uint8)
@@ -69,6 +70,7 @@ class cuda_test(unittest.TestCase):
self.assertTrue(cuMat.step == 0)
self.assertTrue(cuMat.size() == (0, 0))
+ @unittest.skipIf(cv.cuda.getCudaEnabledDeviceCount() == 0, "No cuda device found")
def test_cuda_denoising(self):
self.assertEqual(True, hasattr(cv.cuda, 'fastNlMeansDenoising'))