Skip to content

Commit 356c487

Browse files
committed
added test_intersection_contains
1 parent 76ffdd9 commit 356c487

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

tests/unit/date/test_daterange.py

+16
Original file line numberDiff line numberDiff line change
@@ -234,3 +234,19 @@ def test_intersection_preserves_boundaries():
234234
assert DateRange('20110101', '20110102', OPEN_OPEN) == DateRange('20110101', '20110102', CLOSED_OPEN).intersection(DateRange('20110101', '20110102', OPEN_OPEN))
235235
assert DateRange('20110101', '20110102', OPEN_OPEN) == DateRange('20110101', '20110102', OPEN_OPEN).intersection(DateRange('20110101', '20110102', OPEN_CLOSED))
236236

237+
238+
def test_intersection_contains():
239+
# assert ((d in dr1) & (d in dr2)) == (d in (dr1 & dr2)) for any interval combination
240+
start, end = dt(2018,1,1), dt(2018,1,2)
241+
date_ranges = [DateRange(start, end, interval) for interval in CLOSED_CLOSED.__class__]
242+
243+
def equal_contains(date,dr1,dr2):
244+
return ((date in dr1) and (date in dr2)) == (date in dr1.intersection(dr2))
245+
246+
assert all(equal_contains(start,dr1,dr2) for dr1 in date_ranges for dr2 in date_ranges)
247+
assert all(equal_contains(end,dr1,dr2) for dr1 in date_ranges for dr2 in date_ranges)
248+
249+
250+
251+
252+

0 commit comments

Comments
 (0)