| 62 | ) |
| 63 | @settings(deadline=10000) |
| 64 | def test_allcompare(self, d, n, num_procs): |
| 65 | dims = [] |
| 66 | for _ in range(d): |
| 67 | dims.append(np.random.randint(1, high=n)) |
| 68 | test_data = np.random.ranf(size=tuple(dims)).astype(np.float32) |
| 69 | |
| 70 | with TemporaryDirectory() as tempdir: |
| 71 | processes = [] |
| 72 | for idx in range(num_procs): |
| 73 | process = Process( |
| 74 | target=allcompare_process, |
| 75 | args=(tempdir, idx, test_data, num_procs) |
| 76 | ) |
| 77 | processes.append(process) |
| 78 | process.start() |
| 79 | |
| 80 | while len(processes) > 0: |
| 81 | process = processes.pop() |
| 82 | process.join() |
| 83 | |
| 84 | |
| 85 | if __name__ == "__main__": |