summaryrefslogtreecommitdiffstats
path: root/tests/duration/test_arithmetic.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/duration/test_arithmetic.py')
-rw-r--r--tests/duration/test_arithmetic.py85
1 files changed, 85 insertions, 0 deletions
diff --git a/tests/duration/test_arithmetic.py b/tests/duration/test_arithmetic.py
new file mode 100644
index 0000000..cba4d39
--- /dev/null
+++ b/tests/duration/test_arithmetic.py
@@ -0,0 +1,85 @@
+from __future__ import annotations
+
+import pendulum
+
+from tests.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(days=2, seconds=35, microseconds=522222)
+ mul = it / 1.1
+
+ assert isinstance(mul, pendulum.Duration)
+ assert_duration(mul, 0, 0, 0, 1, 19, 38, 43, 202020)
+
+ 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)
+
+ it = pendulum.duration(years=2, months=4, days=2, seconds=35, microseconds=522222)
+ mul = it / 2.0
+
+ 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)