Python2017/labs02/test_task.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-


def suma(a, b):
    return a + b

def tests(f):
    inputs = [(2, 3), (0, 0), (1, 1)]
    outputs = [5, 0, 2]

    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(suma))