1 files changed, 51 insertions, 0 deletions
diff --git a/tests/period/test_arithmetic.py b/tests/period/test_arithmetic.py
new file mode 100644
index 0000000..47b6ddf
--- /dev/null
+++ b/tests/period/test_arithmetic.py
@@ -0,0 +1,51 @@
+import pendulum
+
+from ..conftest import assert_duration
+
+
+def test_multiply():
+    dt1 = pendulum.DateTime(2016, 8, 7, 12, 34, 56)
+    dt2 = dt1.add(days=6, seconds=34)
+    it = pendulum.period(dt1, dt2)
+    mul = it * 2
+    assert isinstance(mul, pendulum.Duration)
+    assert_duration(mul, 0, 0, 1, 5, 0, 1, 8)
+
+    dt1 = pendulum.DateTime(2016, 8, 7, 12, 34, 56)
+    dt2 = dt1.add(days=6, seconds=34)
+    it = pendulum.period(dt1, dt2)
+    mul = it * 2
+    assert isinstance(mul, pendulum.Duration)
+    assert_duration(mul, 0, 0, 1, 5, 0, 1, 8)
+
+
+def test_divide():
+    dt1 = pendulum.DateTime(2016, 8, 7, 12, 34, 56)
+    dt2 = dt1.add(days=2, seconds=34)
+    it = pendulum.period(dt1, dt2)
+    mul = it / 2
+    assert isinstance(mul, pendulum.Duration)
+    assert_duration(mul, 0, 0, 0, 1, 0, 0, 17)
+
+    dt1 = pendulum.DateTime(2016, 8, 7, 12, 34, 56)
+    dt2 = dt1.add(days=2, seconds=35)
+    it = pendulum.period(dt1, dt2)
+    mul = it / 2
+    assert isinstance(mul, pendulum.Duration)
+    assert_duration(mul, 0, 0, 0, 1, 0, 0, 17)
+
+
+def test_floor_divide():
+    dt1 = pendulum.DateTime(2016, 8, 7, 12, 34, 56)
+    dt2 = dt1.add(days=2, seconds=34)
+    it = pendulum.period(dt1, dt2)
+    mul = it // 2
+    assert isinstance(mul, pendulum.Duration)
+    assert_duration(mul, 0, 0, 0, 1, 0, 0, 17)
+
+    dt1 = pendulum.DateTime(2016, 8, 7, 12, 34, 56)
+    dt2 = dt1.add(days=2, seconds=35)
+    it = pendulum.period(dt1, dt2)
+    mul = it // 3
+    assert isinstance(mul, pendulum.Duration)
+    assert_duration(mul, 0, 0, 0, 0, 16, 0, 11)