use LooseVersion

Author: simonjayhawkins

pandas/tests/extension/test_numpy.py

@@ -1,3 +1,5 @@
+from distutils.version import LooseVersion
+
 import numpy as np
 import pytest
 
@@ -8,8 +10,6 @@
 
 from . import base
 
-NP_VERSION_INFO = tuple(int(x) for x in np.__version__.split('.')[:2])
-
 
 @pytest.fixture(params=['float', 'object'])
 def dtype(request):
@@ -50,7 +50,7 @@ def data_missing(allow_in_pandas, dtype):
     # For NumPy <1.16, np.array([np.nan, (1,)]) raises
     # ValueError: setting an array element with a sequence.
     if dtype.numpy_dtype == 'object':
-        if NP_VERSION_INFO < (1, 16):
+        if LooseVersion(np.__version__) < LooseVersion("1.16"):
             raise pytest.skip("Skipping for NumPy <1.16")
         return PandasArray(np.array([np.nan, (1,)]))
     return PandasArray(np.array([np.nan, 1.0]))