From 0369c3fb861c478b22a49d4693441d37b10daa61 Mon Sep 17 00:00:00 2001 From: Dorian Pula Date: Tue, 5 Nov 2019 22:35:04 -0500 Subject: [PATCH] Separate out the batching code. --- batch_test.py | 22 ++++++++++++++++++++++ test.py | 15 --------------- 2 files changed, 22 insertions(+), 15 deletions(-) create mode 100644 batch_test.py diff --git a/batch_test.py b/batch_test.py new file mode 100644 index 0000000..1f9dd3b --- /dev/null +++ b/batch_test.py @@ -0,0 +1,22 @@ +import random + +import pytest + +import unit_converter + + +@pytest.fixture(scope="module") +def batch_numbers(): + return [random.random() * 100 for x in range(1000)] + + +def batch_python_unit_converter(temperatures): + return [celsius * 1.8 + 32.0 for celsius in temperatures] + + +def test_using_python_batch(benchmark, batch_numbers): + benchmark(batch_python_unit_converter, batch_numbers) + + +def test_using_rust_batch(benchmark, batch_numbers): + benchmark(unit_converter.batch_convert_celsius_to_fahrenheit, batch_numbers) diff --git a/test.py b/test.py index 4750bae..05191e0 100644 --- a/test.py +++ b/test.py @@ -1,15 +1,8 @@ import random -import pytest - import unit_converter -@pytest.fixture(scope="module") -def batch_numbers(): - return [random.random() * 100 for x in range(1000)] - - def python_unit_converter(celsius): return celsius * 1.8 + 32.0 @@ -43,11 +36,3 @@ def test_using_rust(benchmark): temperature = random.random() * 100 result = benchmark(unit_converter.convert_celsius_to_fahrenheit, temperature) assert round(result, 3) == round(python_unit_converter(temperature), 3) - - -def test_using_python_batch(benchmark, batch_numbers): - benchmark(batch_python_unit_converter, batch_numbers) - - -def test_using_rust_batch(benchmark, batch_numbers): - benchmark(unit_converter.batch_convert_celsius_to_fahrenheit, batch_numbers)