From b8fd59c95e0b65b62afb05f4dce226e7dfda3dad Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Perceval=20Wajsb=C3=BCrt?= Date: Mon, 4 Dec 2023 17:28:32 +0100 Subject: [PATCH] test: skip koalas test if not installed --- tests/conftest.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index 8526cae8b..e002f45c3 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,6 +1,7 @@ from datetime import datetime import pandas as pd +import pytest import spacy from pytest import fixture @@ -151,7 +152,6 @@ def df_notes(): def make_df_note(text, module): - import databricks.koalas # noqa F401 from pyspark.sql import types as T from pyspark.sql.session import SparkSession @@ -180,6 +180,10 @@ def make_df_note(text, module): return notes if module == "koalas": + try: + import databricks.koalas # noqa F401 + except ImportError: + pytest.skip("Koalas not installed") return notes.to_koalas()