@@ -33,21 +33,6 @@ def dirpath(datapath):
33
33
return datapath ("io" , "data" , "orc" )
34
34
35
35
36
- # setup engines & skips
37
- @pytest .fixture (
38
- params = [
39
- pytest .param (
40
- "pyarrow" ,
41
- marks = pytest .mark .skipif (
42
- not _HAVE_PYARROW , reason = "pyarrow is not installed"
43
- ),
44
- )
45
- ]
46
- )
47
- def engine (request ):
48
- return request .param
49
-
50
-
51
36
@pytest .fixture
52
37
def pa ():
53
38
if not _HAVE_PYARROW :
@@ -76,7 +61,7 @@ def test_invalid_engine(dirpath):
76
61
read_orc (inputfile , engine = engine , columns = ["boolean1" ])
77
62
78
63
79
- def test_orc_reader_empty (dirpath , engine ):
64
+ def test_orc_reader_empty (dirpath , pa ):
80
65
columns = [
81
66
"boolean1" ,
82
67
"byte1" ,
@@ -104,12 +89,12 @@ def test_orc_reader_empty(dirpath, engine):
104
89
expected [colname ] = pd .Series (dtype = dtype )
105
90
106
91
inputfile = os .path .join (dirpath , "TestOrcFile.emptyFile.orc" )
107
- got = read_orc (inputfile , engine = engine , columns = columns )
92
+ got = read_orc (inputfile , columns = columns )
108
93
109
94
tm .assert_equal (expected , got )
110
95
111
96
112
- def test_orc_reader_basic (dirpath , engine ):
97
+ def test_orc_reader_basic (dirpath , pa ):
113
98
data = {
114
99
"boolean1" : np .array ([False , True ], dtype = "bool" ),
115
100
"byte1" : np .array ([1 , 100 ], dtype = "int8" ),
@@ -124,12 +109,12 @@ def test_orc_reader_basic(dirpath, engine):
124
109
expected = pd .DataFrame .from_dict (data )
125
110
126
111
inputfile = os .path .join (dirpath , "TestOrcFile.test1.orc" )
127
- got = read_orc (inputfile , engine = engine , columns = data .keys ())
112
+ got = read_orc (inputfile , columns = data .keys ())
128
113
129
114
tm .assert_equal (expected , got )
130
115
131
116
132
- def test_orc_reader_decimal (dirpath , engine ):
117
+ def test_orc_reader_decimal (dirpath , pa ):
133
118
from decimal import Decimal
134
119
135
120
# Only testing the first 10 rows of data
@@ -153,12 +138,12 @@ def test_orc_reader_decimal(dirpath, engine):
153
138
expected = pd .DataFrame .from_dict (data )
154
139
155
140
inputfile = os .path .join (dirpath , "TestOrcFile.decimal.orc" )
156
- got = read_orc (inputfile , engine = engine ).iloc [:10 ]
141
+ got = read_orc (inputfile ).iloc [:10 ]
157
142
158
143
tm .assert_equal (expected , got )
159
144
160
145
161
- def test_orc_reader_date_low (dirpath , engine ):
146
+ def test_orc_reader_date_low (dirpath , pa ):
162
147
data = {
163
148
"time" : np .array (
164
149
[
@@ -194,12 +179,12 @@ def test_orc_reader_date_low(dirpath, engine):
194
179
expected = pd .DataFrame .from_dict (data )
195
180
196
181
inputfile = os .path .join (dirpath , "TestOrcFile.testDate1900.orc" )
197
- got = read_orc (inputfile , engine = engine ).iloc [:10 ]
182
+ got = read_orc (inputfile ).iloc [:10 ]
198
183
199
184
tm .assert_equal (expected , got )
200
185
201
186
202
- def test_orc_reader_date_high (dirpath , engine ):
187
+ def test_orc_reader_date_high (dirpath , pa ):
203
188
data = {
204
189
"time" : np .array (
205
190
[
@@ -235,12 +220,12 @@ def test_orc_reader_date_high(dirpath, engine):
235
220
expected = pd .DataFrame .from_dict (data )
236
221
237
222
inputfile = os .path .join (dirpath , "TestOrcFile.testDate2038.orc" )
238
- got = read_orc (inputfile , engine = engine ).iloc [:10 ]
223
+ got = read_orc (inputfile ).iloc [:10 ]
239
224
240
225
tm .assert_equal (expected , got )
241
226
242
227
243
- def test_orc_reader_snappy_compressed (dirpath , engine ):
228
+ def test_orc_reader_snappy_compressed (dirpath , pa ):
244
229
data = {
245
230
"int1" : np .array (
246
231
[
@@ -276,6 +261,6 @@ def test_orc_reader_snappy_compressed(dirpath, engine):
276
261
expected = pd .DataFrame .from_dict (data )
277
262
278
263
inputfile = os .path .join (dirpath , "TestOrcFile.testSnappy.orc" )
279
- got = read_orc (inputfile , engine = engine ).iloc [:10 ]
264
+ got = read_orc (inputfile ).iloc [:10 ]
280
265
281
266
tm .assert_equal (expected , got )
0 commit comments