from datetime import time
import locale

import numpy as np
import pytest

import pandas as pd
import pandas._testing as tm
from pandas.core.tools.times import to_time


class TestToTime:
    @pytest.mark.parametrize(
        "time_string",
        [
            "14:15",
            "1415",
            "2:15pm",
            "0215pm",
            "14:15:00",
            "141500",
            "2:15:00pm",
            "021500pm",
            "2:15 pm",
            "0215 pm",
            "2:15:00 pm",
            "021500 pm",
            time(14, 15),
        ],
    )
    def test_parsers_time(self, time_string):
        with tm.set_locale("C", locale.LC_TIME):
            # GH#11818
            assert to_time(time_string) == time(14, 15)

    def test_parsers_time_space_before_meridiem(self):
        with tm.set_locale("C", locale.LC_TIME):
            # GH#18793 the space before AM/PM used to make these unparsable
            arg = ["3:25:00 AM", "3:25:00 PM", "12:30:00 AM", "12:30:00 PM"]
            expected = [time(3, 25), time(15, 25), time(0, 30), time(12, 30)]
            assert to_time(arg) == expected
            assert to_time(arg, infer_time_format=True) == expected

    def test_odd_format(self):
        new_string = "14.15"
        assert to_time(new_string, format="%H.%M") == time(14, 15)

    def test_arraylike(self):
        arg = ["14:15", "20:20"]
        expected_arr = [time(14, 15), time(20, 20)]
        assert to_time(arg) == expected_arr
        assert to_time(arg, format="%H:%M") == expected_arr
        assert to_time(arg, infer_time_format=True) == expected_arr
        assert to_time(arg, format="%I:%M%p", errors="coerce") == [None, None]

        with pytest.raises(ValueError, match="errors must be"):
            to_time(arg, format="%I:%M%p", errors="ignore")

        msg = "Cannot convert.+to a time with given format"
        with pytest.raises(ValueError, match=msg):
            to_time(arg, format="%I:%M%p", errors="raise")

        tm.assert_series_equal(
            to_time(pd.Series(arg, name="test")), pd.Series(expected_arr, name="test")
        )

        res = to_time(np.array(arg))
        assert isinstance(res, list)
        assert res == expected_arr