From 42d0d485a990e484c92cd71fc76a7438a32f55a2 Mon Sep 17 00:00:00 2001 From: olgavrou Date: Mon, 11 Sep 2023 13:33:43 -0400 Subject: [PATCH] black formatting --- libs/experimental/tests/unit_tests/__init__.py | 3 ++- libs/experimental/tests/unit_tests/test_data_anonymizer.py | 2 +- .../tests/unit_tests/test_reversible_data_anonymizer.py | 1 + 3 files changed, 4 insertions(+), 2 deletions(-) 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():