Skip to content

Commit b0c1671

Browse files
jorisvandenbosschenickleus27
authored andcommitted
TST/COMPAT: update csv test to infer time with pyarrow>=6.0 (pandas-dev#44381)
1 parent c3cefa7 commit b0c1671

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

pandas/compat/pyarrow.py

+2
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,11 @@
1212
pa_version_under3p0 = _palv < Version("3.0.0")
1313
pa_version_under4p0 = _palv < Version("4.0.0")
1414
pa_version_under5p0 = _palv < Version("5.0.0")
15+
pa_version_under6p0 = _palv < Version("6.0.0")
1516
except ImportError:
1617
pa_version_under1p0 = True
1718
pa_version_under2p0 = True
1819
pa_version_under3p0 = True
1920
pa_version_under4p0 = True
2021
pa_version_under5p0 = True
22+
pa_version_under6p0 = True

pandas/tests/io/parser/test_parse_dates.py

+6
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
is_platform_windows,
2727
np_array_datetime64_compat,
2828
)
29+
from pandas.compat.pyarrow import pa_version_under6p0
2930

3031
import pandas as pd
3132
from pandas import (
@@ -431,6 +432,11 @@ def test_date_col_as_index_col(all_parsers):
431432
columns=["X0", "X2", "X3", "X4", "X5", "X6", "X7"],
432433
index=index,
433434
)
435+
if parser.engine == "pyarrow" and not pa_version_under6p0:
436+
# https://github.com/pandas-dev/pandas/issues/44231
437+
# pyarrow 6.0 starts to infer time type
438+
expected["X2"] = pd.to_datetime("1970-01-01" + expected["X2"]).dt.time
439+
434440
tm.assert_frame_equal(result, expected)
435441

436442

0 commit comments

Comments
 (0)