Simplify to remove getattr(pa, "array", None) call

Co-Authored-By: Dongdong Tian <[email protected]>
GenericMappingTools · Nov 15, 2024 · 8172102 · 8172102
1 parent edb3438
commit 8172102
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/pygmt/tests/test_clib_virtualfile_from_vectors.py b/pygmt/tests/test_clib_virtualfile_from_vectors.py
@@ -15,8 +15,10 @@
 
 try:
     import pyarrow as pa
+
+    pa_array = pa.array
 except ImportError:
-    pa = None
+    pa_array = None
 
 
 @pytest.fixture(scope="module", name="dtypes")
@@ -65,8 +67,8 @@ def test_virtualfile_from_vectors(dtypes):
         pytest.param(np.array, {"dtype": np.str_}, id="str"),
         pytest.param(np.array, {"dtype": np.object_}, id="object"),
         pytest.param(
-            getattr(pa, "array", None),
-            {},  # pa.string()
+            pa_array,
+            {},  # {"type": pa.string()}
             marks=skip_if_no(package="pyarrow"),
             id="pyarrow",
         ),

diff --git a/pygmt/tests/test_text.py b/pygmt/tests/test_text.py
@@ -13,8 +13,10 @@
 
 try:
     import pyarrow as pa
+
+    pa_array = pa.array
 except ImportError:
-    pa = None
+    pa_array = None
 
 TEST_DATA_DIR = Path(__file__).parent / "data"
 POINTS_DATA = TEST_DATA_DIR / "points.txt"
@@ -60,11 +62,7 @@ def test_text_single_line_of_text(region, projection):
     [
         list,
         pytest.param(np.array, id="numpy"),
-        pytest.param(
-            getattr(pa, "array", None),
-            marks=skip_if_no(package="pyarrow"),
-            id="pyarrow",
-        ),
+        pytest.param(pa_array, marks=skip_if_no(package="pyarrow"), id="pyarrow"),
     ],
 )
 def test_text_multiple_lines_of_text(region, projection, array_func):