SEDONA-725 Add pyflink to Sedona.

Author: Imbruced

python/tests/flink/conftest.py

@@ -2,35 +2,38 @@
 
 import pytest
 
-try:
-    from sedona.flink import SedonaContext
-except ImportError:
-    pass
-
-from pyflink.datastream import StreamExecutionEnvironment
-from pyflink.table import EnvironmentSettings, StreamTableEnvironment
-
 
 EXTRA_JARS = os.getenv("SEDONA_PYFLINK_EXTRA_JARS")
 
 
-@pytest.fixture(scope="module")
-def flink_settings():
-    return EnvironmentSettings.in_streaming_mode()
-
+def has_pyflink():
+    try:
+        import pyflink
+    except ImportError:
+        return False
+    return True
 
-@pytest.fixture(scope="module")
-def stream_env() -> StreamExecutionEnvironment:
-    env = StreamExecutionEnvironment.get_execution_environment()
-    jars = EXTRA_JARS.split(",") if EXTRA_JARS else []
-    for jar in jars:
-        env.add_jars(f"file://{jar}")
 
-    return env
-
-
-@pytest.fixture(scope="module")
-def table_env(
-    stream_env: StreamExecutionEnvironment, flink_settings: EnvironmentSettings
-) -> StreamTableEnvironment:
-    return SedonaContext.create(stream_env, flink_settings)
+if has_pyflink():
+    from sedona.flink import SedonaContext
+    from pyflink.datastream import StreamExecutionEnvironment
+    from pyflink.table import EnvironmentSettings, StreamTableEnvironment
+
+    @pytest.fixture(scope="module")
+    def flink_settings():
+        return EnvironmentSettings.in_streaming_mode()
+
+    @pytest.fixture(scope="module")
+    def stream_env() -> StreamExecutionEnvironment:
+        env = StreamExecutionEnvironment.get_execution_environment()
+        jars = EXTRA_JARS.split(",") if EXTRA_JARS else []
+        for jar in jars:
+            env.add_jars(f"file://{jar}")
+
+        return env
+
+    @pytest.fixture(scope="module")
+    def table_env(
+        stream_env: StreamExecutionEnvironment, flink_settings: EnvironmentSettings
+    ) -> StreamTableEnvironment:
+        return SedonaContext.create(stream_env, flink_settings)