Compare commits
5
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
221ab2c20b | ||
|
|
d9326deeb5 | ||
|
|
e15648232f | ||
|
|
ee994a3938 | ||
|
|
7ae4018e59 |
@@ -319,13 +319,6 @@ def cycle_week_index(
|
|||||||
return (days // 7) % schedule.cycle_length_weeks
|
return (days // 7) % schedule.cycle_length_weeks
|
||||||
|
|
||||||
|
|
||||||
#: Search cap for :meth:`PresenceEngine.next_present_dates`: two full
|
|
||||||
#: years of daily checks. A child absent every day (an all-absent
|
|
||||||
#: pattern with a blocking override) would otherwise loop forever; the
|
|
||||||
#: cap bounds the scan and surfaces the situation as an error instead.
|
|
||||||
MAX_PREVIEW_SCAN_DAYS = 366 * 2
|
|
||||||
|
|
||||||
|
|
||||||
class PresenceEngine:
|
class PresenceEngine:
|
||||||
"""Answers "is this child at this house on this date?".
|
"""Answers "is this child at this house on this date?".
|
||||||
|
|
||||||
@@ -431,45 +424,6 @@ class PresenceEngine:
|
|||||||
week = cycle_week_index(schedule, target)
|
week = cycle_week_index(schedule, target)
|
||||||
return target.weekday() in schedule.pattern[week]
|
return target.weekday() in schedule.pattern[week]
|
||||||
|
|
||||||
def next_present_dates(
|
|
||||||
self,
|
|
||||||
child_id: object,
|
|
||||||
start_date: object,
|
|
||||||
count: int,
|
|
||||||
) -> list[datetime.date]:
|
|
||||||
"""Return the next ``count`` dates the child is present.
|
|
||||||
|
|
||||||
Walks forward from ``start_date`` (INCLUSIVE — a parent
|
|
||||||
previewing "starting Monday" expects Monday itself when the
|
|
||||||
child is present) checking :meth:`is_present` for every day,
|
|
||||||
so overrides and the pattern are honoured identically. The
|
|
||||||
admin Schedule tab renders this preview so a parent can confirm
|
|
||||||
a custody pattern is right before saving it.
|
|
||||||
|
|
||||||
A child absent every scanned day raises ValueError after the
|
|
||||||
two-year search cap — an always-absent child would otherwise
|
|
||||||
loop forever. ``count`` must be a positive integer.
|
|
||||||
"""
|
|
||||||
_validate_child_id(child_id)
|
|
||||||
start = _parse_date(start_date, "start_date")
|
|
||||||
if isinstance(count, bool) or not isinstance(count, int):
|
|
||||||
raise ValueError(f"count must be an integer, got {count!r}")
|
|
||||||
if count < 1:
|
|
||||||
raise ValueError(f"count must be at least 1, got {count}")
|
|
||||||
found: list[datetime.date] = []
|
|
||||||
day = start
|
|
||||||
for _ in range(MAX_PREVIEW_SCAN_DAYS):
|
|
||||||
if self.is_present(child_id, day):
|
|
||||||
found.append(day)
|
|
||||||
if len(found) == count:
|
|
||||||
return found
|
|
||||||
day += datetime.timedelta(days=1)
|
|
||||||
raise ValueError(
|
|
||||||
f"child {child_id} is not present on any of the "
|
|
||||||
f"{MAX_PREVIEW_SCAN_DAYS} days from {start.isoformat()}; "
|
|
||||||
f"{count} present day(s) cannot be previewed"
|
|
||||||
)
|
|
||||||
|
|
||||||
def __setattr__(self, name: object, value: object) -> None:
|
def __setattr__(self, name: object, value: object) -> None:
|
||||||
raise AttributeError(
|
raise AttributeError(
|
||||||
"PresenceEngine is an immutable snapshot; build a new engine "
|
"PresenceEngine is an immutable snapshot; build a new engine "
|
||||||
|
|||||||
@@ -599,216 +599,3 @@ def test_engine_override_lists_are_frozen() -> None:
|
|||||||
engine._overrides[1].append(entry)
|
engine._overrides[1].append(entry)
|
||||||
with pytest.raises(AttributeError, match="immutable snapshot"):
|
with pytest.raises(AttributeError, match="immutable snapshot"):
|
||||||
engine._overrides = {}
|
engine._overrides = {}
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# next_present_dates: the parent's pattern preview
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
def _preview_engine() -> PresenceEngine:
|
|
||||||
"""Declan: home all of week 0, away all of week 1 (anchor
|
|
||||||
2026-01-05, a Monday); an away override Tue Jan 13 (his away week)
|
|
||||||
would matter only for the override test."""
|
|
||||||
return PresenceEngine(
|
|
||||||
{
|
|
||||||
1: PresenceSchedule(
|
|
||||||
1, 2, datetime.date(2026, 1, 5),
|
|
||||||
{0: {0, 1, 2, 3, 4, 5, 6}, 1: set()},
|
|
||||||
)
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def test_preview_returns_the_next_present_days_of_a_known_pattern() -> None:
|
|
||||||
engine = _preview_engine()
|
|
||||||
# Starting Monday Jan 5 (present week 0): five straight days.
|
|
||||||
preview = engine.next_present_dates(1, "2026-01-05", 5)
|
|
||||||
assert preview == [
|
|
||||||
datetime.date(2026, 1, 5),
|
|
||||||
datetime.date(2026, 1, 6),
|
|
||||||
datetime.date(2026, 1, 7),
|
|
||||||
datetime.date(2026, 1, 8),
|
|
||||||
datetime.date(2026, 1, 9),
|
|
||||||
]
|
|
||||||
# Starting mid-week: the rest of week 0 first, then the absent
|
|
||||||
# week 1 is skipped and the next present week picks up.
|
|
||||||
preview = engine.next_present_dates(1, "2026-01-07", 3)
|
|
||||||
assert preview == [
|
|
||||||
datetime.date(2026, 1, 7),
|
|
||||||
datetime.date(2026, 1, 8),
|
|
||||||
datetime.date(2026, 1, 9),
|
|
||||||
]
|
|
||||||
preview = engine.next_present_dates(1, "2026-01-07", 6)
|
|
||||||
# Week 0 runs Mon Jan 5 .. Sun Jan 11 (all present); the away week
|
|
||||||
# (Jan 12-18) is skipped; week 2 resumes Jan 19.
|
|
||||||
assert preview == [
|
|
||||||
datetime.date(2026, 1, 7),
|
|
||||||
datetime.date(2026, 1, 8),
|
|
||||||
datetime.date(2026, 1, 9),
|
|
||||||
datetime.date(2026, 1, 10),
|
|
||||||
datetime.date(2026, 1, 11),
|
|
||||||
datetime.date(2026, 1, 19),
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
def test_preview_start_date_is_inclusive() -> None:
|
|
||||||
engine = _preview_engine()
|
|
||||||
# Jan 12 (Monday of the away week) is absent: the first present day
|
|
||||||
# is Jan 19 — the start date itself when present.
|
|
||||||
preview = engine.next_present_dates(1, "2026-01-19", 1)
|
|
||||||
assert preview == [datetime.date(2026, 1, 19)]
|
|
||||||
|
|
||||||
|
|
||||||
def test_preview_honours_overrides() -> None:
|
|
||||||
schedule = PresenceSchedule(
|
|
||||||
1, 2, datetime.date(2026, 1, 5),
|
|
||||||
{0: {0, 1, 2, 3, 4, 5, 6}, 1: set()},
|
|
||||||
)
|
|
||||||
engine = PresenceEngine(
|
|
||||||
{1: schedule},
|
|
||||||
overrides={
|
|
||||||
1: [
|
|
||||||
PresenceOverride(
|
|
||||||
1, "2026-01-05", "2026-01-06", False, note="holiday"
|
|
||||||
)
|
|
||||||
]
|
|
||||||
},
|
|
||||||
)
|
|
||||||
# Jan 5-6 blocked by the override: the preview starts Jan 7.
|
|
||||||
preview = engine.next_present_dates(1, "2026-01-04", 3)
|
|
||||||
assert preview == [
|
|
||||||
datetime.date(2026, 1, 7),
|
|
||||||
datetime.date(2026, 1, 8),
|
|
||||||
datetime.date(2026, 1, 9),
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
def test_preview_schedule_free_child_returns_consecutive_days() -> None:
|
|
||||||
engine = PresenceEngine({})
|
|
||||||
preview = engine.next_present_dates(3, "2026-03-01", 3)
|
|
||||||
assert preview == [
|
|
||||||
datetime.date(2026, 3, 1),
|
|
||||||
datetime.date(2026, 3, 2),
|
|
||||||
datetime.date(2026, 3, 3),
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
def test_preview_rejects_bad_inputs() -> None:
|
|
||||||
engine = _preview_engine()
|
|
||||||
with pytest.raises(ValueError, match="count must be an integer"):
|
|
||||||
engine.next_present_dates(1, "2026-01-05", "3")
|
|
||||||
with pytest.raises(ValueError, match="count must be at least 1"):
|
|
||||||
engine.next_present_dates(1, "2026-01-05", 0)
|
|
||||||
with pytest.raises(ValueError, match="child_id must be an integer"):
|
|
||||||
engine.next_present_dates(True, "2026-01-05", 1)
|
|
||||||
with pytest.raises(ValueError, match="plain calendar date"):
|
|
||||||
engine.next_present_dates(1, datetime.datetime(2026, 1, 5), 1)
|
|
||||||
|
|
||||||
|
|
||||||
def test_preview_always_absent_child_raises_not_loops() -> None:
|
|
||||||
"""A child absent every scanned day must surface as an error, not
|
|
||||||
hang the admin preview."""
|
|
||||||
engine = PresenceEngine(
|
|
||||||
{
|
|
||||||
1: PresenceSchedule(
|
|
||||||
1, 1, datetime.date(2026, 1, 5), {0: set()}
|
|
||||||
)
|
|
||||||
}
|
|
||||||
)
|
|
||||||
with pytest.raises(ValueError, match="not present on any"):
|
|
||||||
engine.next_present_dates(1, "2026-01-05", 2)
|
|
||||||
|
|
||||||
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
# The week-parity regression guard (Feature 05, D-004)
|
|
||||||
# ---------------------------------------------------------------------------
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
("year", "length", "reason"),
|
|
||||||
[
|
|
||||||
(2020, 366, "leap year, 53 ISO weeks"),
|
|
||||||
(2015, 365, "365 days, 53 ISO weeks"),
|
|
||||||
],
|
|
||||||
)
|
|
||||||
def test_custody_rotation_walks_full_years_without_inversion(
|
|
||||||
year, length, reason
|
|
||||||
) -> None:
|
|
||||||
"""REGRESSION GUARD for the week-parity bug (D-004).
|
|
||||||
|
|
||||||
Walks a two-week alternating rotation day by day across a FULL
|
|
||||||
{length}-day year ({year} — {reason}) plus the following January,
|
|
||||||
asserting the present/absent sequence never breaks or inverts at
|
|
||||||
any point, including January 1st. ISO week parity silently flips
|
|
||||||
at the ISO-week rollover (the Monday on or after January 4th —
|
|
||||||
January 4th itself in both of these years, which end ISO week 53)
|
|
||||||
and would invert every custody schedule there; anchor-date
|
|
||||||
arithmetic must not.
|
|
||||||
""".format(length=length, year=year, reason=reason)
|
|
||||||
if year == 2020:
|
|
||||||
assert datetime.date(2020, 12, 31).isocalendar()[1] == 53
|
|
||||||
assert (datetime.date(2021, 1, 1) - datetime.date(2020, 1, 1)).days == 366
|
|
||||||
if year == 2015:
|
|
||||||
assert datetime.date(2015, 12, 31).isocalendar()[1] == 53
|
|
||||||
assert (datetime.date(2016, 1, 1) - datetime.date(2015, 1, 1)).days == 365
|
|
||||||
|
|
||||||
# The first Monday of January (2020: Jan 6; 2015: Jan 5) — an
|
|
||||||
# on-week anchor, whatever the year.
|
|
||||||
anchor = datetime.date(year, 1, 1) + datetime.timedelta(
|
|
||||||
days=(7 - datetime.date(year, 1, 1).weekday()) % 7
|
|
||||||
)
|
|
||||||
assert anchor.weekday() == 0
|
|
||||||
rotation = PresenceSchedule(
|
|
||||||
1, 2, anchor, {0: {0, 1, 2, 3, 4, 5, 6}, 1: set()}
|
|
||||||
)
|
|
||||||
engine = PresenceEngine({1: rotation})
|
|
||||||
|
|
||||||
# Walk EVERY day of the year. The expectation is derived from the
|
|
||||||
# anchor's 7-day BLOCKS, not from the engine: each block of seven
|
|
||||||
# days starting at the anchor is uniformly present (even block
|
|
||||||
# index) or uniformly absent (odd block index), and consecutive
|
|
||||||
# blocks alternate. A parity-based implementation resets on
|
|
||||||
# January 1st and breaks either the uniformity of a block or the
|
|
||||||
# alternation between them — most visibly in the block spanning
|
|
||||||
# December 31st -> January 1st of a 53-ISO-week year.
|
|
||||||
# Walk the FULL year AND the first two weeks of the following
|
|
||||||
# January: the block spanning December 31st -> January 1st must
|
|
||||||
# stay uniform ACROSS the boundary, so an implementation that
|
|
||||||
# resets when the target year differs from the anchor's fails
|
|
||||||
# here (the post-boundary days of its final block disagree).
|
|
||||||
first_day = datetime.date(year, 1, 1)
|
|
||||||
last_day = datetime.date(year + 1, 1, 14)
|
|
||||||
days_seen = 0
|
|
||||||
day = first_day
|
|
||||||
while day <= last_day:
|
|
||||||
days_since_anchor = (day - anchor).days
|
|
||||||
block = days_since_anchor // 7 # which 7-day block of the cycle
|
|
||||||
block_start = anchor + datetime.timedelta(days=block * 7)
|
|
||||||
# Uniformity within the block: every day of this 7-day block
|
|
||||||
# must answer exactly like the block's first day.
|
|
||||||
expected = ((block % 2) == 0)
|
|
||||||
assert engine.is_present(1, day) is expected, (
|
|
||||||
f"{day} (block {block}, day offset "
|
|
||||||
f"{(day - block_start).days}): present="
|
|
||||||
f"{engine.is_present(1, day)}, expected {expected}"
|
|
||||||
)
|
|
||||||
# The block's OTHER days must agree — including the days on
|
|
||||||
# the far side of January 1st.
|
|
||||||
for offset in range(7):
|
|
||||||
block_day = block_start + datetime.timedelta(days=offset)
|
|
||||||
if first_day <= block_day <= last_day:
|
|
||||||
assert engine.is_present(1, block_day) is expected, (
|
|
||||||
f"{block_day} disagrees with its block {block}"
|
|
||||||
)
|
|
||||||
days_seen += 1
|
|
||||||
day += datetime.timedelta(days=1)
|
|
||||||
# The walked year itself contributed exactly its full length.
|
|
||||||
year_days = sum(
|
|
||||||
1
|
|
||||||
for offset in range((last_day - first_day).days + 1)
|
|
||||||
if (first_day + datetime.timedelta(days=offset)).year == year
|
|
||||||
)
|
|
||||||
assert year_days == length, (
|
|
||||||
f"walked {year_days} days of {year}, expected {length}"
|
|
||||||
)
|
|
||||||
|
|||||||
Reference in New Issue
Block a user