diff --git a/libs/experimental/tests/unit_tests/__init__.py b/libs/experimental/tests/unit_tests/__init__.py index 0594a6bd14..f177990421 100644 --- a/libs/experimental/tests/unit_tests/__init__.py +++ b/libs/experimental/tests/unit_tests/__init__.py @@ -1,8 +1,9 @@ import ctypes + def is_libcublas_available() -> bool: try: ctypes.CDLL("libcublas.so") return True except OSError: - return False \ No newline at end of file + return False diff --git a/libs/experimental/tests/unit_tests/test_data_anonymizer.py b/libs/experimental/tests/unit_tests/test_data_anonymizer.py index 07f1519488..17ea8ff1a5 100644 --- a/libs/experimental/tests/unit_tests/test_data_anonymizer.py +++ b/libs/experimental/tests/unit_tests/test_data_anonymizer.py @@ -12,6 +12,7 @@ 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(): @@ -19,7 +20,6 @@ def check_libcublas() -> Iterator[None]: yield - @pytest.mark.requires("presidio_analyzer", "presidio_anonymizer", "faker") @pytest.mark.parametrize( "analyzed_fields,should_contain", 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 e2cc1912c2..4302cec992 100644 --- a/libs/experimental/tests/unit_tests/test_reversible_data_anonymizer.py +++ b/libs/experimental/tests/unit_tests/test_reversible_data_anonymizer.py @@ -13,6 +13,7 @@ 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():