From f0a4499f968b6bae65084073ba42a2943ab9d756 Mon Sep 17 00:00:00 2001 From: Ivan Enderlin Date: Sat, 13 Apr 2019 16:27:05 -0700 Subject: [PATCH] test(benchmarks) Lower `N`. --- benchmarks/test_nbody.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/benchmarks/test_nbody.py b/benchmarks/test_nbody.py index 1018f066..2dc64537 100644 --- a/benchmarks/test_nbody.py +++ b/benchmarks/test_nbody.py @@ -9,12 +9,14 @@ here = os.path.dirname(os.path.realpath(__file__)) TEST_BYTES = open(here + '/nbody.wasm', 'rb').read() +N = 5000 + def test_benchmark_nbody_with_wasmer(benchmark): instance = wasmer.Instance(TEST_BYTES) nbody = instance.exports.main def bench(): - return nbody(5000000) + return nbody(N) assert benchmark(bench) @@ -23,7 +25,7 @@ def test_benchmark_nbody_with_ppci(benchmark): nbody = instance.exports.main def bench(): - return nbody(5000000) + return nbody(N) assert benchmark(bench) @@ -116,7 +118,7 @@ def report_energy(bodies=SYSTEM, pairs=PAIRS, e=0.0): e -= (m1 * m2) / ((dx * dx + dy * dy + dz * dz) ** 0.5) for (r, [vx, vy, vz], m) in bodies: e += m * (vx * vx + vy * vy + vz * vz) / 2. - print("%.9f" % e) + return e def offset_momentum(ref, bodies=SYSTEM, px=0.0, py=0.0, pz=0.0): for (r, [vx, vy, vz], m) in bodies: @@ -132,9 +134,9 @@ def nbody(n, ref='sun'): offset_momentum(BODIES[ref]) report_energy() advance(0.01, n) - report_energy() + return report_energy() def bench(): - return nbody(5000000) + return nbody(N) assert benchmark(bench)