diff options
Diffstat (limited to '')
-rw-r--r-- | tests/duration/test_arithmetic.py | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/tests/duration/test_arithmetic.py b/tests/duration/test_arithmetic.py new file mode 100644 index 0000000..fad38b6 --- /dev/null +++ b/tests/duration/test_arithmetic.py @@ -0,0 +1,71 @@ +import pendulum + +from ..conftest import assert_duration + + +def test_multiply(): + it = pendulum.duration(days=6, seconds=34, microseconds=522222) + mul = it * 2 + + assert isinstance(mul, pendulum.Duration) + assert_duration(mul, 0, 0, 1, 5, 0, 1, 9, 44444) + + it = pendulum.duration(days=6, seconds=34, microseconds=522222) + mul = 2 * it + + assert isinstance(mul, pendulum.Duration) + assert_duration(mul, 0, 0, 1, 5, 0, 1, 9, 44444) + + it = pendulum.duration( + years=2, months=3, weeks=4, days=6, seconds=34, microseconds=522222 + ) + mul = 2 * it + + assert isinstance(mul, pendulum.Duration) + assert_duration(mul, 4, 6, 9, 5, 0, 1, 9, 44444) + + +def test_divide(): + it = pendulum.duration(days=2, seconds=34, microseconds=522222) + mul = it / 2 + + assert isinstance(mul, pendulum.Duration) + assert_duration(mul, 0, 0, 0, 1, 0, 0, 17, 261111) + + it = pendulum.duration(days=2, seconds=35, microseconds=522222) + mul = it / 2 + + assert isinstance(mul, pendulum.Duration) + assert_duration(mul, 0, 0, 0, 1, 0, 0, 17, 761111) + + it = pendulum.duration(years=2, months=4, days=2, seconds=35, microseconds=522222) + mul = it / 2 + + assert isinstance(mul, pendulum.Duration) + assert_duration(mul, 1, 2, 0, 1, 0, 0, 17, 761111) + + +def test_floor_divide(): + it = pendulum.duration(days=2, seconds=34, microseconds=522222) + mul = it // 2 + + assert isinstance(mul, pendulum.Duration) + assert_duration(mul, 0, 0, 0, 1, 0, 0, 17, 261111) + + it = pendulum.duration(days=2, seconds=35, microseconds=522222) + mul = it // 3 + + assert isinstance(mul, pendulum.Duration) + assert_duration(mul, 0, 0, 0, 0, 16, 0, 11, 840740) + + it = pendulum.duration(years=2, months=4, days=2, seconds=34, microseconds=522222) + mul = it // 2 + + assert isinstance(mul, pendulum.Duration) + assert_duration(mul, 1, 2, 0, 1, 0, 0, 17, 261111) + + it = pendulum.duration(years=2, months=4, days=2, seconds=35, microseconds=522222) + mul = it // 3 + + assert isinstance(mul, pendulum.Duration) + assert_duration(mul, 0, 1, 0, 0, 16, 0, 11, 840740) |