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)