test_date_converters.py

from datetime import datetime

import numpy as np

import pandas.util.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)])

    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])
    result = conv.parse_date_fields(years, months, days)

    expected = np.array([datetime(2007, 1, 3), datetime(2008, 2, 4)])
    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])

    result = conv.parse_all_fields(years, months, days,
                                   hours, minutes, seconds)
    expected = np.array([datetime(2007, 1, 3, 5, 7, 9),
                         datetime(2008, 2, 4, 6, 8, 0)])
    tm.assert_numpy_array_equal(result, expected)