test_chrono.py 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209
  1. # -*- coding: utf-8 -*-
  2. from pybind11_tests import chrono as m
  3. import datetime
  4. import pytest
  5. import env # noqa: F401
  6. def test_chrono_system_clock():
  7. # Get the time from both c++ and datetime
  8. date0 = datetime.datetime.today()
  9. date1 = m.test_chrono1()
  10. date2 = datetime.datetime.today()
  11. # The returned value should be a datetime
  12. assert isinstance(date1, datetime.datetime)
  13. # The numbers should vary by a very small amount (time it took to execute)
  14. diff_python = abs(date2 - date0)
  15. diff = abs(date1 - date2)
  16. # There should never be a days difference
  17. assert diff.days == 0
  18. # Since datetime.datetime.today() calls time.time(), and on some platforms
  19. # that has 1 second accuracy, we compare this way
  20. assert diff.seconds <= diff_python.seconds
  21. def test_chrono_system_clock_roundtrip():
  22. date1 = datetime.datetime.today()
  23. # Roundtrip the time
  24. date2 = m.test_chrono2(date1)
  25. # The returned value should be a datetime
  26. assert isinstance(date2, datetime.datetime)
  27. # They should be identical (no information lost on roundtrip)
  28. diff = abs(date1 - date2)
  29. assert diff == datetime.timedelta(0)
  30. def test_chrono_system_clock_roundtrip_date():
  31. date1 = datetime.date.today()
  32. # Roundtrip the time
  33. datetime2 = m.test_chrono2(date1)
  34. date2 = datetime2.date()
  35. time2 = datetime2.time()
  36. # The returned value should be a datetime
  37. assert isinstance(datetime2, datetime.datetime)
  38. assert isinstance(date2, datetime.date)
  39. assert isinstance(time2, datetime.time)
  40. # They should be identical (no information lost on roundtrip)
  41. diff = abs(date1 - date2)
  42. assert diff.days == 0
  43. assert diff.seconds == 0
  44. assert diff.microseconds == 0
  45. # Year, Month & Day should be the same after the round trip
  46. assert date1 == date2
  47. # There should be no time information
  48. assert time2.hour == 0
  49. assert time2.minute == 0
  50. assert time2.second == 0
  51. assert time2.microsecond == 0
  52. SKIP_TZ_ENV_ON_WIN = pytest.mark.skipif(
  53. "env.WIN", reason="TZ environment variable only supported on POSIX"
  54. )
  55. @pytest.mark.parametrize(
  56. "time1",
  57. [
  58. datetime.datetime.today().time(),
  59. datetime.time(0, 0, 0),
  60. datetime.time(0, 0, 0, 1),
  61. datetime.time(0, 28, 45, 109827),
  62. datetime.time(0, 59, 59, 999999),
  63. datetime.time(1, 0, 0),
  64. datetime.time(5, 59, 59, 0),
  65. datetime.time(5, 59, 59, 1),
  66. ],
  67. )
  68. @pytest.mark.parametrize(
  69. "tz",
  70. [
  71. None,
  72. pytest.param("Europe/Brussels", marks=SKIP_TZ_ENV_ON_WIN),
  73. pytest.param("Asia/Pyongyang", marks=SKIP_TZ_ENV_ON_WIN),
  74. pytest.param("America/New_York", marks=SKIP_TZ_ENV_ON_WIN),
  75. ],
  76. )
  77. def test_chrono_system_clock_roundtrip_time(time1, tz, monkeypatch):
  78. if tz is not None:
  79. monkeypatch.setenv("TZ", "/usr/share/zoneinfo/{}".format(tz))
  80. # Roundtrip the time
  81. datetime2 = m.test_chrono2(time1)
  82. date2 = datetime2.date()
  83. time2 = datetime2.time()
  84. # The returned value should be a datetime
  85. assert isinstance(datetime2, datetime.datetime)
  86. assert isinstance(date2, datetime.date)
  87. assert isinstance(time2, datetime.time)
  88. # Hour, Minute, Second & Microsecond should be the same after the round trip
  89. assert time1 == time2
  90. # There should be no date information (i.e. date = python base date)
  91. assert date2.year == 1970
  92. assert date2.month == 1
  93. assert date2.day == 1
  94. def test_chrono_duration_roundtrip():
  95. # Get the difference between two times (a timedelta)
  96. date1 = datetime.datetime.today()
  97. date2 = datetime.datetime.today()
  98. diff = date2 - date1
  99. # Make sure this is a timedelta
  100. assert isinstance(diff, datetime.timedelta)
  101. cpp_diff = m.test_chrono3(diff)
  102. assert cpp_diff == diff
  103. # Negative timedelta roundtrip
  104. diff = datetime.timedelta(microseconds=-1)
  105. cpp_diff = m.test_chrono3(diff)
  106. assert cpp_diff == diff
  107. def test_chrono_duration_subtraction_equivalence():
  108. date1 = datetime.datetime.today()
  109. date2 = datetime.datetime.today()
  110. diff = date2 - date1
  111. cpp_diff = m.test_chrono4(date2, date1)
  112. assert cpp_diff == diff
  113. def test_chrono_duration_subtraction_equivalence_date():
  114. date1 = datetime.date.today()
  115. date2 = datetime.date.today()
  116. diff = date2 - date1
  117. cpp_diff = m.test_chrono4(date2, date1)
  118. assert cpp_diff == diff
  119. def test_chrono_steady_clock():
  120. time1 = m.test_chrono5()
  121. assert isinstance(time1, datetime.timedelta)
  122. def test_chrono_steady_clock_roundtrip():
  123. time1 = datetime.timedelta(days=10, seconds=10, microseconds=100)
  124. time2 = m.test_chrono6(time1)
  125. assert isinstance(time2, datetime.timedelta)
  126. # They should be identical (no information lost on roundtrip)
  127. assert time1 == time2
  128. def test_floating_point_duration():
  129. # Test using a floating point number in seconds
  130. time = m.test_chrono7(35.525123)
  131. assert isinstance(time, datetime.timedelta)
  132. assert time.seconds == 35
  133. assert 525122 <= time.microseconds <= 525123
  134. diff = m.test_chrono_float_diff(43.789012, 1.123456)
  135. assert diff.seconds == 42
  136. assert 665556 <= diff.microseconds <= 665557
  137. def test_nano_timepoint():
  138. time = datetime.datetime.now()
  139. time1 = m.test_nano_timepoint(time, datetime.timedelta(seconds=60))
  140. assert time1 == time + datetime.timedelta(seconds=60)
  141. def test_chrono_different_resolutions():
  142. resolutions = m.different_resolutions()
  143. time = datetime.datetime.now()
  144. resolutions.timestamp_h = time
  145. resolutions.timestamp_m = time
  146. resolutions.timestamp_s = time
  147. resolutions.timestamp_ms = time
  148. resolutions.timestamp_us = time