Add assert to rangesum JUUUUUUUST in case

Signed-off-by: Alek Ratzloff <alekratz@gmail.com>
This commit is contained in:
2021-12-07 11:47:27 -08:00
parent f4351ce9a7
commit eaf69cabeb

View File

@@ -4,6 +4,7 @@ from typing import Sequence
def rangesum(rn: range):
assert rn.step == 1, "this function doesn't work when range step != 1"
return (((rn.stop + 1) * rn.stop) // 2) - (((rn.start + 1) * rn.start) // 2)