Skip to content

Commit 9390e0a

Browse files
committed
remove magic number
1 parent bee9b3e commit 9390e0a

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/test_multiprocessing.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,11 +58,12 @@ def test_multiple_processes():
5858
# multiple processes
5959
pool = Pool()
6060
num_jobs = 10
61-
solutions = pool.starmap(odr_fit, [case1, case2, case3]*num_jobs)
61+
cases = [case1, case2, case3]
62+
solutions = pool.starmap(odr_fit, cases*num_jobs)
6263
pool.close()
6364
pool.join()
6465

65-
for i in range(0, len(solutions), 3):
66+
for i in range(0, len(solutions), len(cases)):
6667
assert np.allclose(solutions[i].beta, sol1.beta)
6768
assert np.allclose(solutions[i+1].beta, sol2.beta)
6869
assert np.allclose(solutions[i+2].beta, sol3.beta)

0 commit comments

Comments
 (0)