Python2017/labs02/task04.py
2017-11-27 21:23:07 +01:00

28 lines
623 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):
if n >= 1:
return sum(range(1,n+1))
else:
return 0
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))