From 7185fdc9906294e234f172f5b089546b00784059 Mon Sep 17 00:00:00 2001
From: olgavrou <olgavrou@gmail.com>
Date: Mon, 11 Sep 2023 13:26:41 -0400
Subject: [PATCH] check if libcublas is available before running extended tests

---
 libs/experimental/tests/unit_tests/__init__.py            | 8 ++++++++
 .../experimental/tests/unit_tests/test_data_anonymizer.py | 8 ++++++++
 .../tests/unit_tests/test_reversible_data_anonymizer.py   | 7 +++++++
 3 files changed, 23 insertions(+)

diff --git a/libs/experimental/tests/unit_tests/__init__.py b/libs/experimental/tests/unit_tests/__init__.py
index e69de29bb2..48e02109f0 100644
--- a/libs/experimental/tests/unit_tests/__init__.py
+++ b/libs/experimental/tests/unit_tests/__init__.py
@@ -0,0 +1,8 @@
+import ctypes
+
+def is_libcublas_available():
+    try:
+        ctypes.CDLL("libcublas.so")
+        return True
+    except OSError:
+        return False
\ No newline at end of file
diff --git a/libs/experimental/tests/unit_tests/test_data_anonymizer.py b/libs/experimental/tests/unit_tests/test_data_anonymizer.py
index 138b60eca8..07f1519488 100644
--- a/libs/experimental/tests/unit_tests/test_data_anonymizer.py
+++ b/libs/experimental/tests/unit_tests/test_data_anonymizer.py
@@ -1,6 +1,7 @@
 from typing import Iterator, List
 
 import pytest
+from . import is_libcublas_available
 
 
 @pytest.fixture(scope="module", autouse=True)
@@ -11,6 +12,13 @@ def check_spacy_model() -> Iterator[None]:
         pytest.skip(reason="Spacy model 'en_core_web_lg' not installed")
     yield
 
+@pytest.fixture(scope="module", autouse=True)
+def check_libcublas() -> Iterator[None]:
+    if not is_libcublas_available():
+        pytest.skip(reason="libcublas.so is not available")
+    yield
+
+
 
 @pytest.mark.requires("presidio_analyzer", "presidio_anonymizer", "faker")
 @pytest.mark.parametrize(
diff --git a/libs/experimental/tests/unit_tests/test_reversible_data_anonymizer.py b/libs/experimental/tests/unit_tests/test_reversible_data_anonymizer.py
index 9484a0e9dc..e2cc1912c2 100644
--- a/libs/experimental/tests/unit_tests/test_reversible_data_anonymizer.py
+++ b/libs/experimental/tests/unit_tests/test_reversible_data_anonymizer.py
@@ -2,6 +2,7 @@ import os
 from typing import Iterator, List
 
 import pytest
+from . import is_libcublas_available
 
 
 @pytest.fixture(scope="module", autouse=True)
@@ -12,6 +13,12 @@ def check_spacy_model() -> Iterator[None]:
         pytest.skip(reason="Spacy model 'en_core_web_lg' not installed")
     yield
 
+@pytest.fixture(scope="module", autouse=True)
+def check_libcublas() -> Iterator[None]:
+    if not is_libcublas_available():
+        pytest.skip(reason="libcublas.so is not available")
+    yield
+
 
 @pytest.mark.requires("presidio_analyzer", "presidio_anonymizer", "faker")
 @pytest.mark.parametrize(