fix numpy test setup

Author: jorisvandenbossche

pandas/tests/extension/test_numpy.py

@@ -77,8 +77,10 @@ def allow_in_pandas(monkeypatch):
 @pytest.fixture
 def data(allow_in_pandas, dtype):
     if dtype.numpy_dtype == "object":
-        return pd.Series([(i,) for i in range(100)])._values
-    return NumpyExtensionArray(np.arange(1, 101, dtype=dtype._dtype))
+        arr = pd.Series([(i,) for i in range(100)])._values
+    else:
+        arr = np.arange(1, 101, dtype=dtype._dtype)
+    return NumpyExtensionArray(arr)
 
 
 @pytest.fixture