|
31 | 31 | import dpnp |
32 | 32 | import numpy as np |
33 | 33 |
|
34 | | -ROUNDS=30 |
35 | | -ITERATIONS=4 |
| 34 | +ROUNDS = 30 |
| 35 | +ITERATIONS = 4 |
| 36 | + |
| 37 | +NNUMBERS = 2**26 |
36 | 38 |
|
37 | | -NNUMBERS=2**26 |
38 | 39 |
|
39 | 40 | @pytest.mark.parametrize("function", [dpnp.random.beta, np.random.beta], |
40 | | - ids=["dpnp", "numpy"]) |
| 41 | + ids=["dpnp", "numpy"]) |
41 | 42 | def test_beta(benchmark, function): |
42 | 43 | result = benchmark.pedantic(target=function, args=(4.0, 5.0, NNUMBERS,), |
43 | 44 | rounds=ROUNDS, iterations=ITERATIONS) |
44 | 45 |
|
45 | 46 |
|
46 | 47 | @pytest.mark.parametrize("function", [dpnp.random.exponential, np.random.exponential], |
47 | | - ids=["dpnp", "numpy"]) |
| 48 | + ids=["dpnp", "numpy"]) |
48 | 49 | def test_exponential(benchmark, function): |
49 | 50 | result = benchmark.pedantic(target=function, args=(4.0, NNUMBERS,), |
50 | 51 | rounds=ROUNDS, iterations=ITERATIONS) |
51 | 52 |
|
52 | 53 |
|
53 | 54 | @pytest.mark.parametrize("function", [dpnp.random.gamma, np.random.gamma], |
54 | | - ids=["dpnp", "numpy"]) |
| 55 | + ids=["dpnp", "numpy"]) |
55 | 56 | def test_gamma(benchmark, function): |
56 | 57 | result = benchmark.pedantic(target=function, args=(2.0, 4.0, NNUMBERS,), |
57 | 58 | rounds=ROUNDS, iterations=ITERATIONS) |
58 | 59 |
|
59 | 60 |
|
60 | 61 | @pytest.mark.parametrize("function", [dpnp.random.normal, np.random.normal], |
61 | | - ids=["dpnp", "numpy"]) |
| 62 | + ids=["dpnp", "numpy"]) |
62 | 63 | def test_normal(benchmark, function): |
63 | 64 | result = benchmark.pedantic(target=function, args=(0.0, 1.0, NNUMBERS,), |
64 | 65 | rounds=ROUNDS, iterations=ITERATIONS) |
65 | 66 |
|
66 | 67 |
|
67 | 68 | @pytest.mark.parametrize("function", [dpnp.random.uniform, np.random.uniform], |
68 | | - ids=["dpnp", "numpy"]) |
| 69 | + ids=["dpnp", "numpy"]) |
69 | 70 | def test_uniform(benchmark, function): |
70 | 71 | result = benchmark.pedantic(target=function, args=(0.0, 1.0, NNUMBERS,), |
71 | 72 | rounds=ROUNDS, iterations=ITERATIONS) |
0 commit comments