From b4254e7f91afb6863c13c6b4b1914e6f5e287a3c Mon Sep 17 00:00:00 2001 From: shotat Date: Sun, 30 May 2021 20:57:40 +0900 Subject: add benchmark --- tests/benchmark/test_api.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 tests/benchmark/test_api.py (limited to 'tests/benchmark') diff --git a/tests/benchmark/test_api.py b/tests/benchmark/test_api.py new file mode 100644 index 00000000..11698cb0 --- /dev/null +++ b/tests/benchmark/test_api.py @@ -0,0 +1,28 @@ +import pytest +from isort import api + +imperfect_content = "import b\nimport a\n" +fixed_content = "import a\nimport b\n" + + +@pytest.fixture +def imperfect(tmpdir) -> None: + imperfect_file = tmpdir.join("test_needs_changes.py") + imperfect_file.write_text(imperfect_content, "utf8") + return imperfect_file + + +def test_sort_file(benchmark, imperfect) -> None: + def sort_file(): + api.sort_file(imperfect) + + benchmark.pedantic(sort_file, iterations=10, rounds=100) + assert imperfect.read() == fixed_content + + +def test_sort_file_in_place(benchmark, imperfect) -> None: + def sort_file(): + api.sort_file(imperfect, overwrite_in_place=True) + + benchmark.pedantic(sort_file, iterations=10, rounds=100) + assert imperfect.read() == fixed_content -- cgit v1.2.1