test_chrono.py 5.56 KB
Newer Older
1
import datetime
2

3
4
5
import pytest

import env  # noqa: F401
6
from pybind11_tests import chrono as m
7
8
9
10


def test_chrono_system_clock():
    # Get the time from both c++ and datetime
11
    date0 = datetime.datetime.today()
12
    date1 = m.test_chrono1()
13
14
15
16
17
18
    date2 = datetime.datetime.today()

    # The returned value should be a datetime
    assert isinstance(date1, datetime.datetime)

    # The numbers should vary by a very small amount (time it took to execute)
19
    diff_python = abs(date2 - date0)
20
21
    diff = abs(date1 - date2)

22
    # There should never be a days difference
23
24
    assert diff.days == 0

25
    # Since datetime.datetime.today() calls time.time(), and on some platforms
26
27
    # that has 1 second accuracy, we compare this way
    assert diff.seconds <= diff_python.seconds
28
29
30
31
32
33


def test_chrono_system_clock_roundtrip():
    date1 = datetime.datetime.today()

    # Roundtrip the time
34
    date2 = m.test_chrono2(date1)
35
36
37
38
39
40

    # The returned value should be a datetime
    assert isinstance(date2, datetime.datetime)

    # They should be identical (no information lost on roundtrip)
    diff = abs(date1 - date2)
41
    assert diff == datetime.timedelta(0)
42
43


44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
def test_chrono_system_clock_roundtrip_date():
    date1 = datetime.date.today()

    # Roundtrip the time
    datetime2 = m.test_chrono2(date1)
    date2 = datetime2.date()
    time2 = datetime2.time()

    # The returned value should be a datetime
    assert isinstance(datetime2, datetime.datetime)
    assert isinstance(date2, datetime.date)
    assert isinstance(time2, datetime.time)

    # They should be identical (no information lost on roundtrip)
    diff = abs(date1 - date2)
    assert diff.days == 0
    assert diff.seconds == 0
    assert diff.microseconds == 0

    # Year, Month & Day should be the same after the round trip
64
    assert date1 == date2
65
66
67
68
69
70
71
72

    # There should be no time information
    assert time2.hour == 0
    assert time2.minute == 0
    assert time2.second == 0
    assert time2.microsecond == 0


73
74
75
76
77
SKIP_TZ_ENV_ON_WIN = pytest.mark.skipif(
    "env.WIN", reason="TZ environment variable only supported on POSIX"
)


78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
@pytest.mark.parametrize(
    "time1",
    [
        datetime.datetime.today().time(),
        datetime.time(0, 0, 0),
        datetime.time(0, 0, 0, 1),
        datetime.time(0, 28, 45, 109827),
        datetime.time(0, 59, 59, 999999),
        datetime.time(1, 0, 0),
        datetime.time(5, 59, 59, 0),
        datetime.time(5, 59, 59, 1),
    ],
)
@pytest.mark.parametrize(
    "tz",
    [
        None,
        pytest.param("Europe/Brussels", marks=SKIP_TZ_ENV_ON_WIN),
        pytest.param("Asia/Pyongyang", marks=SKIP_TZ_ENV_ON_WIN),
        pytest.param("America/New_York", marks=SKIP_TZ_ENV_ON_WIN),
    ],
)
100
101
def test_chrono_system_clock_roundtrip_time(time1, tz, monkeypatch):
    if tz is not None:
102
        monkeypatch.setenv("TZ", f"/usr/share/zoneinfo/{tz}")
103
104
105
106
107
108
109
110
111
112
113
114

    # Roundtrip the time
    datetime2 = m.test_chrono2(time1)
    date2 = datetime2.date()
    time2 = datetime2.time()

    # The returned value should be a datetime
    assert isinstance(datetime2, datetime.datetime)
    assert isinstance(date2, datetime.date)
    assert isinstance(time2, datetime.time)

    # Hour, Minute, Second & Microsecond should be the same after the round trip
115
    assert time1 == time2
116
117
118
119
120
121
122

    # There should be no date information (i.e. date = python base date)
    assert date2.year == 1970
    assert date2.month == 1
    assert date2.day == 1


123
def test_chrono_duration_roundtrip():
124
    # Get the difference between two times (a timedelta)
125
126
127
128
129
130
131
    date1 = datetime.datetime.today()
    date2 = datetime.datetime.today()
    diff = date2 - date1

    # Make sure this is a timedelta
    assert isinstance(diff, datetime.timedelta)

132
    cpp_diff = m.test_chrono3(diff)
133

134
135
136
137
138
139
140
    assert cpp_diff == diff

    # Negative timedelta roundtrip
    diff = datetime.timedelta(microseconds=-1)
    cpp_diff = m.test_chrono3(diff)

    assert cpp_diff == diff
141
142
143
144
145
146
147


def test_chrono_duration_subtraction_equivalence():
    date1 = datetime.datetime.today()
    date2 = datetime.datetime.today()

    diff = date2 - date1
148
    cpp_diff = m.test_chrono4(date2, date1)
149

150
    assert cpp_diff == diff
151
152


153
154
155
156
157
158
159
def test_chrono_duration_subtraction_equivalence_date():
    date1 = datetime.date.today()
    date2 = datetime.date.today()

    diff = date2 - date1
    cpp_diff = m.test_chrono4(date2, date1)

160
    assert cpp_diff == diff
161
162


163
def test_chrono_steady_clock():
164
    time1 = m.test_chrono5()
165
    assert isinstance(time1, datetime.timedelta)
166
167
168


def test_chrono_steady_clock_roundtrip():
169
    time1 = datetime.timedelta(days=10, seconds=10, microseconds=100)
170
    time2 = m.test_chrono6(time1)
171

172
    assert isinstance(time2, datetime.timedelta)
173
174

    # They should be identical (no information lost on roundtrip)
175
    assert time1 == time2
176
177
178


def test_floating_point_duration():
179
180
    # Test using a floating point number in seconds
    time = m.test_chrono7(35.525123)
181
182
183
184

    assert isinstance(time, datetime.timedelta)

    assert time.seconds == 35
185
    assert 525122 <= time.microseconds <= 525123
186

187
    diff = m.test_chrono_float_diff(43.789012, 1.123456)
188
189
    assert diff.seconds == 42
    assert 665556 <= diff.microseconds <= 665557
190
191
192
193
194


def test_nano_timepoint():
    time = datetime.datetime.now()
    time1 = m.test_nano_timepoint(time, datetime.timedelta(seconds=60))
195
    assert time1 == time + datetime.timedelta(seconds=60)
196
197
198
199
200
201
202
203
204
205


def test_chrono_different_resolutions():
    resolutions = m.different_resolutions()
    time = datetime.datetime.now()
    resolutions.timestamp_h = time
    resolutions.timestamp_m = time
    resolutions.timestamp_s = time
    resolutions.timestamp_ms = time
    resolutions.timestamp_us = time