GH-39444: [Python] Fix parquet import in encryption test (#40505)

jorisvandenbossche · web-flow · commit fd1e9ca81f4d · 2024-03-14T08:48:47.000+01:00
Small follow-up on #39623 fixing am import in the test. But first testing here if it actually fails. * GitHub Issue: #39444 Authored-by: Joris Van den Bossche <jorisvandenbossche@gmail.com> Signed-off-by: Joris Van den Bossche <jorisvandenbossche@gmail.com>
diff --git a/dev/tasks/tasks.yml b/dev/tasks/tasks.yml
@@ -82,6 +82,7 @@ groups:
 
   python:
     - test-*python*
+    - example-*python*
 
   r:
     - test*-r-*
diff --git a/python/pyarrow/tests/test_dataset_encryption.py b/python/pyarrow/tests/test_dataset_encryption.py
@@ -20,14 +20,16 @@
 import numpy as np
 import pyarrow.fs as fs
 import pyarrow as pa
-import pyarrow.parquet as pq
+
 import pytest
 
 encryption_unavailable = False
 
 try:
+    import pyarrow.parquet as pq
     import pyarrow.dataset as ds
 except ImportError:
+    pq = None
     ds = None
 
 try:
