Cross-attention mask omits the source interval limit · case 01

"""Failure Map reference implementation. Python standard library only."""
import json

N = 1
observations = []
def solve(source_count, query_count, offset, left, right):
    return [[True for _ in range(source_count)] for _ in range(query_count)]
def check(label, actual, expected):
    observations.append({"check": label, "actual": actual, "expected": expected, "passed": actual == expected})
check('moving source window',solve(N+3,3,1,1,0),[[max(0,q+1-1)<=s<min(N+3,q+2) for s in range(N+3)] for q in range(3)])
check('one-to-one',solve(3,3,0,0,0),[[s==q for s in range(3)] for q in range(3)])
check('empty source',solve(0,2,0,1,1),[[],[]])
check('empty queries',solve(3,0,0,1,1),[])
check('window clipped left',solve(3,1,0,2,0),[[True,False,False]])
check('window clipped right',solve(3,3,0,0,1)[2],[False,False,True])
print(json.dumps({"observations": observations, "passed": all(x["passed"] for x in observations)}, ensure_ascii=False))
raise SystemExit(0 if all(x["passed"] for x in observations) else 1)

SHA-256 / 3eb4104b50656a4d8fafb789fb3009c99c066e3cdffe5f61126c949c5bf1d31a

"""Failure Map reference implementation. Python standard library only."""
import json

N = 1
observations = []
def solve(source_count, query_count, offset, left, right):
    return [[max(0,offset-left)<=s<min(source_count,offset+right+1) for s in range(source_count)] for q in range(query_count)]
def check(label, actual, expected):
    observations.append({"check": label, "actual": actual, "expected": expected, "passed": actual == expected})
check('moving source window',solve(N+3,3,1,1,0),[[max(0,q+1-1)<=s<min(N+3,q+2) for s in range(N+3)] for q in range(3)])
check('one-to-one',solve(3,3,0,0,0),[[s==q for s in range(3)] for q in range(3)])
check('empty source',solve(0,2,0,1,1),[[],[]])
check('empty queries',solve(3,0,0,1,1),[])
check('window clipped left',solve(3,1,0,2,0),[[True,False,False]])
check('window clipped right',solve(3,3,0,0,1)[2],[False,False,True])
print(json.dumps({"observations": observations, "passed": all(x["passed"] for x in observations)}, ensure_ascii=False))
raise SystemExit(0 if all(x["passed"] for x in observations) else 1)

SHA-256 / b6fbc03bcf39dd231a41429b1cfe133127e3d998fcebd0ba68ea7ce99580d481