File tree Expand file tree Collapse file tree 2 files changed +2
-19
lines changed Expand file tree Collapse file tree 2 files changed +2
-19
lines changed Original file line number Diff line number Diff line change @@ -26,12 +26,9 @@ def fixture_dtypes_pandas(dtypes):
26
26
"""
27
27
List of supported pandas dtypes.
28
28
"""
29
- dtypes_pandas = dtypes .copy ()
30
-
31
29
if find_spec ("pyarrow" ) is not None :
32
- dtypes_pandas .extend ([f"{ dtype } [pyarrow]" for dtype in dtypes_pandas ])
33
-
34
- return tuple (dtypes_pandas )
30
+ dtypes_pandas = [f"{ np .dtype (dtype ).name } [pyarrow]" for dtype in dtypes ]
31
+ return tuple (dtypes + dtypes_pandas )
35
32
36
33
37
34
@pytest .mark .benchmark
Original file line number Diff line number Diff line change 2
2
Test the Session.open_virtualfile method.
3
3
"""
4
4
5
- from importlib .util import find_spec
6
5
from pathlib import Path
7
6
8
7
import numpy as np
@@ -32,19 +31,6 @@ def fixture_dtypes():
32
31
return [dtype for dtype in DTYPES_NUMERIC if dtype != np .timedelta64 ]
33
32
34
33
35
- @pytest .fixture (scope = "module" , name = "dtypes_pandas" )
36
- def fixture_dtypes_pandas (dtypes ):
37
- """
38
- List of supported pandas dtypes.
39
- """
40
- dtypes_pandas = dtypes .copy ()
41
-
42
- if find_spec ("pyarrow" ) is not None :
43
- dtypes_pandas .extend ([f"{ dtype } [pyarrow]" for dtype in dtypes_pandas ])
44
-
45
- return tuple (dtypes_pandas )
46
-
47
-
48
34
@pytest .mark .benchmark
49
35
def test_open_virtualfile (dtypes ):
50
36
"""
You can’t perform that action at this time.
0 commit comments