diff options
Diffstat (limited to 'ext/pybind11/tests/test_chrono.py')
| -rw-r--r-- | ext/pybind11/tests/test_chrono.py | 41 |
1 files changed, 11 insertions, 30 deletions
diff --git a/ext/pybind11/tests/test_chrono.py b/ext/pybind11/tests/test_chrono.py index 55094edbf..2b75bd191 100644 --- a/ext/pybind11/tests/test_chrono.py +++ b/ext/pybind11/tests/test_chrono.py @@ -1,11 +1,11 @@ +from pybind11_tests import chrono as m +import datetime def test_chrono_system_clock(): - from pybind11_tests import test_chrono1 - import datetime # Get the time from both c++ and datetime - date1 = test_chrono1() + date1 = m.test_chrono1() date2 = datetime.datetime.today() # The returned value should be a datetime @@ -25,13 +25,10 @@ def test_chrono_system_clock(): def test_chrono_system_clock_roundtrip(): - from pybind11_tests import test_chrono2 - import datetime - date1 = datetime.datetime.today() # Roundtrip the time - date2 = test_chrono2(date1) + date2 = m.test_chrono2(date1) # The returned value should be a datetime assert isinstance(date2, datetime.datetime) @@ -44,8 +41,6 @@ def test_chrono_system_clock_roundtrip(): def test_chrono_duration_roundtrip(): - from pybind11_tests import test_chrono3 - import datetime # Get the difference between two times (a timedelta) date1 = datetime.datetime.today() @@ -55,7 +50,7 @@ def test_chrono_duration_roundtrip(): # Make sure this is a timedelta assert isinstance(diff, datetime.timedelta) - cpp_diff = test_chrono3(diff) + cpp_diff = m.test_chrono3(diff) assert cpp_diff.days == diff.days assert cpp_diff.seconds == diff.seconds @@ -63,14 +58,12 @@ def test_chrono_duration_roundtrip(): def test_chrono_duration_subtraction_equivalence(): - from pybind11_tests import test_chrono4 - import datetime date1 = datetime.datetime.today() date2 = datetime.datetime.today() diff = date2 - date1 - cpp_diff = test_chrono4(date2, date1) + cpp_diff = m.test_chrono4(date2, date1) assert cpp_diff.days == diff.days assert cpp_diff.seconds == diff.seconds @@ -78,22 +71,13 @@ def test_chrono_duration_subtraction_equivalence(): def test_chrono_steady_clock(): - from pybind11_tests import test_chrono5 - import datetime - - time1 = test_chrono5() - time2 = test_chrono5() - + time1 = m.test_chrono5() assert isinstance(time1, datetime.timedelta) - assert isinstance(time2, datetime.timedelta) def test_chrono_steady_clock_roundtrip(): - from pybind11_tests import test_chrono6 - import datetime - time1 = datetime.timedelta(days=10, seconds=10, microseconds=100) - time2 = test_chrono6(time1) + time2 = m.test_chrono6(time1) assert isinstance(time2, datetime.timedelta) @@ -104,17 +88,14 @@ def test_chrono_steady_clock_roundtrip(): def test_floating_point_duration(): - from pybind11_tests import test_chrono7, test_chrono_float_diff - import datetime - - # Test using 35.525123 seconds as an example floating point number in seconds - time = test_chrono7(35.525123) + # Test using a floating point number in seconds + time = m.test_chrono7(35.525123) assert isinstance(time, datetime.timedelta) assert time.seconds == 35 assert 525122 <= time.microseconds <= 525123 - diff = test_chrono_float_diff(43.789012, 1.123456) + diff = m.test_chrono_float_diff(43.789012, 1.123456) assert diff.seconds == 42 assert 665556 <= diff.microseconds <= 665557 |