Add assert to rangesum JUUUUUUUST in case
Signed-off-by: Alek Ratzloff <alekratz@gmail.com>
This commit is contained in:
@@ -4,6 +4,7 @@ from typing import Sequence
|
|||||||
|
|
||||||
|
|
||||||
def rangesum(rn: range):
|
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)
|
return (((rn.stop + 1) * rn.stop) // 2) - (((rn.start + 1) * rn.start) // 2)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user