map_parallel_mpi_test

Complexity

Total Complexity

1

Size/Duplication

Total Lines	22
Duplicated Lines	0 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
A	f()	0	2	1

#!/usr/bin/env python

from map_parallel import map_parallel
from map_parallel import starmap_parallel

ARGS = [
    [3, 4, 5],
    [5, 12, 13],
    [1, 2, 3]
]


def f(x, y, z):
    return x * x + y * y == z * z


if __name__ == "__main__":
    args = list(map(list, zip(*ARGS)))
    truth = list(map(f, *args))
    res = map_parallel(f, *args, mode='mpi')
    assert res == truth