1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67
|
from __future__ import annotations
import pickle
from datetime import timedelta
import pendulum
def test_pickle():
dt1 = pendulum.datetime(2016, 11, 18)
dt2 = pendulum.datetime(2016, 11, 20)
p = pendulum.interval(dt1, dt2)
s = pickle.dumps(p)
p2 = pickle.loads(s)
assert p.start == p2.start
assert p.end == p2.end
assert p.invert == p2.invert
p = pendulum.interval(dt2, dt1)
s = pickle.dumps(p)
p2 = pickle.loads(s)
assert p.start == p2.start
assert p.end == p2.end
assert p.invert == p2.invert
p = pendulum.interval(dt2, dt1, True)
s = pickle.dumps(p)
p2 = pickle.loads(s)
assert p.start == p2.start
assert p.end == p2.end
assert p.invert == p2.invert
def test_comparison_to_timedelta():
dt1 = pendulum.datetime(2016, 11, 18)
dt2 = pendulum.datetime(2016, 11, 20)
interval = dt2 - dt1
assert interval < timedelta(days=4)
def test_equality_to_timedelta():
dt1 = pendulum.datetime(2016, 11, 18)
dt2 = pendulum.datetime(2016, 11, 20)
interval = dt2 - dt1
assert interval == timedelta(days=2)
def test_inequality():
dt1 = pendulum.datetime(2016, 11, 18)
dt2 = pendulum.datetime(2016, 11, 20)
dt3 = pendulum.datetime(2016, 11, 22)
interval1 = dt2 - dt1
interval2 = dt3 - dt2
interval3 = dt3 - dt1
assert interval1 != interval2
assert interval1 != interval3
|