Python2017/labs02/task04.py
2017-11-27 01:06:55 +01:00

24 lines
615 B
Python

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Napisz funkcję sum_from_one_to_n zwracającą sume liczb od 1 do n.
Jeśli podany argument jest mniejszy od 1 powinna być zwracana wartość 0.
"""
def sum_from_one_to_n(n):
return 0 if n < 1 else reduce(lambda x, y: (x+y), range(1,n+1))
def tests(f):
inputs = [[999], [-100]]
outputs = [499500, 0]
for input, output in zip(inputs, outputs):
if f(*input) != output:
return "ERROR: {}!={}".format(f(*input), output)
break
return "TESTS PASSED"
if __name__ == "__main__":
print(tests(sum_from_one_to_n))