|
4 | 4 | from datetime import datetime as dt
|
5 | 5 | from arctic.date import datetime_to_ms, ms_to_datetime, mktz, to_pandas_closed_closed, DateRange, OPEN_OPEN, CLOSED_CLOSED
|
6 | 6 | from arctic.date._mktz import DEFAULT_TIME_ZONE_NAME
|
| 7 | +from arctic.date._util import to_dt |
7 | 8 |
|
8 | 9 |
|
9 | 10 | @pytest.mark.parametrize('pdt', [
|
@@ -60,3 +61,28 @@ def test_daterange_closedclosed_no_tz():
|
60 | 61 | CLOSED_CLOSED)
|
61 | 62 | act = to_pandas_closed_closed(date_range)
|
62 | 63 | assert act == expected
|
| 64 | + |
| 65 | + |
| 66 | +def test_to_dt_0(): |
| 67 | + assert to_dt(0) == dt(1970, 1, 1, 1, tzinfo=mktz()) |
| 68 | + |
| 69 | + |
| 70 | +def test_to_dt_0_default(): |
| 71 | + assert to_dt(0, mktz('UTC')) == dt(1970, 1, 1, tzinfo=mktz('UTC')) |
| 72 | + |
| 73 | + |
| 74 | +def test_to_dt_dt_no_tz(): |
| 75 | + with pytest.raises(ValueError): |
| 76 | + assert to_dt(dt(1970, 1, 1)) == dt(1970, 1, 1, tzinfo=mktz()) |
| 77 | + |
| 78 | + |
| 79 | +def test_to_dt_dt_no_tz_default(): |
| 80 | + assert to_dt(dt(1970, 1, 1), mktz('UTC')) == dt(1970, 1, 1, tzinfo=mktz('UTC')) |
| 81 | + |
| 82 | + |
| 83 | +def test_to_dt_dt_tz(): |
| 84 | + assert to_dt(dt(1970, 1, 1, tzinfo=mktz('UTC'))) == dt(1970, 1, 1, tzinfo=mktz('UTC')) |
| 85 | + |
| 86 | + |
| 87 | +def test_to_dt_dt_tz_default(): |
| 88 | + assert to_dt(dt(1970, 1, 1, tzinfo=mktz('UTC')), mktz('Europe/London')) == dt(1970, 1, 1, tzinfo=mktz('UTC')) |
0 commit comments