44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
|
from datetime import datetime
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
import pandas._testing as tm
|
||
|
|
|
||
|
|
import pandas.io.date_converters as conv
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_date_time():
|
||
|
|
|
||
|
|
dates = np.array(["2007/1/3", "2008/2/4"], dtype=object)
|
||
|
|
times = np.array(["05:07:09", "06:08:00"], dtype=object)
|
||
|
|
expected = np.array([datetime(2007, 1, 3, 5, 7, 9), datetime(2008, 2, 4, 6, 8, 0)])
|
||
|
|
with tm.assert_produces_warning(FutureWarning):
|
||
|
|
result = conv.parse_date_time(dates, times)
|
||
|
|
tm.assert_numpy_array_equal(result, expected)
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_date_fields():
|
||
|
|
days = np.array([3, 4])
|
||
|
|
months = np.array([1, 2])
|
||
|
|
years = np.array([2007, 2008])
|
||
|
|
expected = np.array([datetime(2007, 1, 3), datetime(2008, 2, 4)])
|
||
|
|
|
||
|
|
with tm.assert_produces_warning(FutureWarning):
|
||
|
|
result = conv.parse_date_fields(years, months, days)
|
||
|
|
tm.assert_numpy_array_equal(result, expected)
|
||
|
|
|
||
|
|
|
||
|
|
def test_parse_all_fields():
|
||
|
|
hours = np.array([5, 6])
|
||
|
|
minutes = np.array([7, 8])
|
||
|
|
seconds = np.array([9, 0])
|
||
|
|
|
||
|
|
days = np.array([3, 4])
|
||
|
|
years = np.array([2007, 2008])
|
||
|
|
months = np.array([1, 2])
|
||
|
|
expected = np.array([datetime(2007, 1, 3, 5, 7, 9), datetime(2008, 2, 4, 6, 8, 0)])
|
||
|
|
|
||
|
|
with tm.assert_produces_warning(FutureWarning):
|
||
|
|
result = conv.parse_all_fields(years, months, days, hours, minutes, seconds)
|
||
|
|
tm.assert_numpy_array_equal(result, expected)
|