perfplot

python-perfplot/README.md

@@ -0,0 +1,23 @@
+# Python `perfplot`: Measuring Performance Made Easy
+
+Here you'll find the code samples and supplementary materials for the corresponding article.
+
+To get set up you should create a virtual environment and install two dependencies:
+
+macOS and Linux:
+
+```shell
+$ python -m venv venv
+$ source venv/bin/activate
+(venv) $ python -m pip install perfplot faker
+```
+
+Windows PowerShell:
+
+```ps
+PS> python -m venv venv
+PS> venv\scripts\activate
+(venv) PS> python -m pip install perfplot faker
+```
+
+From there, you can run any of the scripts included.

python-perfplot/first_match.py

@@ -0,0 +1,33 @@
+import perfplot
+
+
+def build_list(size, fill, value, at_position):
+    return [value if i == at_position else fill for i in range(size)]
+
+
+def find_match_loop(iterable):
+    for value in iterable:
+        if value["population"] > 50:
+            return value
+    return None
+
+
+def find_match_gen(iterable):
+    return next(
+        (value for value in iterable if value["population"] > 50), None
+    )
+
+
+perfplot.show(
+    n_range=[2**n for n in range(25)],
+    setup=lambda n: build_list(
+        size=n,
+        fill={"country": "Nowhere", "population": 10},
+        value={"country": "Atlantis", "population": 100},
+        at_position=n // 2,
+    ),
+    kernels=[find_match_loop, find_match_gen],
+    labels=["loop", "gen"],
+    equality_check=None,
+    relative_to=0,
+)

python-perfplot/generate_random_ints.py

@@ -0,0 +1,12 @@
+import random
+import json
+
+random_ints = [random.randint(0, 10_000) for _ in range(2**25)]
+
+with open("random_ints.txt", mode="w", encoding="utf-8") as file:
+    file.write(json.dumps(random_ints))
+
+reverse_sorted = list(range(2**25, -1, -1))
+
+with open("reverse_sorted.txt", mode="w", encoding="utf-8") as file:
+    file.write(json.dumps(random_ints))

python-perfplot/lambda_vs_getter.py

@@ -0,0 +1,35 @@
+import operator
+import random
+
+import faker
+import perfplot
+
+faker.Faker.seed(10)
+fake = faker.Faker()
+
+
+def create_dict(n):
+    return {fake.name(): random.randint(10, 200) for _ in range(n)}
+
+
+def sort_with_lambda(dict_to_sort: dict):
+    return sorted(dict_to_sort.items(), key=lambda item: item[1])
+
+
+def sort_with_itemgetter(dict_to_sort: dict):
+    sorted(dict_to_sort.items(), key=operator.itemgetter(1))
+
+
+perfplot.live(
+    setup=lambda n: create_dict(n),
+    kernels=[sort_with_lambda, sort_with_itemgetter],
+    n_range=[n**2 for n in range(20)],
+    equality_check=None,
+)
+
+perfplot.live(
+    setup=lambda n: create_dict(n),
+    kernels=[sort_with_lambda, sort_with_itemgetter],
+    n_range=[2**n for n in range(20)],
+    equality_check=None,
+)

python-perfplot/list_files.py

@@ -0,0 +1,66 @@
+import os
+import pathlib
+import shutil
+
+import perfplot
+
+
+def make_dir_with_files(dir_name, number_of_files):
+    flat_dir = pathlib.Path(dir_name)
+
+    if flat_dir.exists():
+        shutil.rmtree(flat_dir)
+
+    flat_dir.mkdir(exist_ok=True)
+
+    for file_number in range(number_of_files):
+        pathlib.Path(flat_dir / pathlib.Path(f"{file_number}.txt")).touch()
+    return flat_dir
+
+
+def glob_star(dir):
+    return list(dir.glob("*"))
+
+
+def glob_star_star(dir):
+    return list(dir.glob("**/*"))
+
+
+def rglob_star(dir):
+    return list(dir.rglob("*"))
+
+
+def iterdir_(dir):
+    return list(dir.iterdir())
+
+
+def os_walk(dir):
+    _, dirs, files = next(os.walk(dir))
+    return [*dirs, *files]
+
+
+def os_scandir(dir):
+    return list(os.scandir(dir))
+
+
+def os_listdir(dir):
+    return os.listdir(dir)
+
+
+perfplot.show(
+    setup=lambda n: make_dir_with_files(
+        "temp",
+        number_of_files=n,
+    ),
+    kernels=[
+        glob_star,
+        glob_star_star,
+        rglob_star,
+        iterdir_,
+        os_walk,
+        os_scandir,
+        os_listdir,
+    ],
+    n_range=[2**n for n in range(13)],
+    equality_check=None,
+)

python-perfplot/perfplot_args_demo.py

@@ -0,0 +1,23 @@
+import time
+
+import perfplot
+
+
+def setup(n):
+    print(f"setting up {n}")
+    return n * "#"
+
+
+def kernel_one(setup_result):
+    print(f"kernel one: {setup_result}")
+    time.sleep(1)
+    return setup_result
+
+
+def kernel_two(setup_result):
+    print(f"kernel two: {setup_result}")
+    time.sleep(1.5)
+    return setup_result
+
+
+perfplot.show(n_range=[1, 2, 3], setup=setup, kernels=[kernel_one, kernel_two])

python-perfplot/plot_sorting_basic.py

@@ -0,0 +1,26 @@
+from random import randint
+
+import perfplot
+
+from sorting import (
+    bubble_sort,
+    insertion_sort,
+    merge_sort,
+    quick_sort,
+    tim_sort,
+    python_built_in_sort,
+)
+
+
+perfplot.show(
+    n_range=range(0, 1_001, 100),
+    setup=lambda n: [randint(0, 1_000) for _ in range(n)],
+    kernels=[
+        bubble_sort,
+        insertion_sort,
+        merge_sort,
+        quick_sort,
+        tim_sort,
+        python_built_in_sort,
+    ],
+)

python-perfplot/plot_sorting_bench.py

@@ -0,0 +1,28 @@
+from random import randint
+
+import perfplot
+
+from sorting import (
+    bubble_sort,
+    insertion_sort,
+    merge_sort,
+    python_built_in_sort,
+    quick_sort,
+    tim_sort,
+)
+
+data = perfplot.bench(
+    n_range=[2**n for n in range(10)],
+    setup=lambda n: [randint(0, 1000) for _ in range(n)],
+    kernels=[
+        bubble_sort,
+        insertion_sort,
+        merge_sort,
+        quick_sort,
+        tim_sort,
+        python_built_in_sort,
+    ],
+)
+
+data.save("sorting_algos_log_x.png", logx=True)
+data.save("sorting_algos_linear_x.png", logx=False)

python-perfplot/plot_sorting_large_range.py

@@ -0,0 +1,26 @@
+from random import randint
+
+import perfplot
+
+from sorting import (
+    bubble_sort,
+    insertion_sort,
+    merge_sort,
+    quick_sort,
+    tim_sort,
+    python_built_in_sort,
+)
+
+perfplot.show(
+    n_range=range(0, 10_001, 1_000),
+    setup=lambda n: [randint(0, 1_000) for _ in range(n)],
+    kernels=[
+        bubble_sort,
+        insertion_sort,
+        merge_sort,
+        quick_sort,
+        tim_sort,
+        python_built_in_sort,
+    ],
+    logy=True,
+)

python-perfplot/plot_sorting_live.py

@@ -0,0 +1,35 @@
+from random import randint
+
+import perfplot
+
+from sorting import (
+    bubble_sort,
+    insertion_sort,
+    merge_sort,
+    quick_sort,
+    tim_sort,
+    python_built_in_sort,
+)
+
+perfplot.live(
+    n_range=[2**n for n in range(15)],
+    setup=lambda n: [randint(0, 1_000) for _ in range(n)],
+    kernels=[
+        bubble_sort,
+        insertion_sort,
+        merge_sort,
+        quick_sort,
+        tim_sort,
+        python_built_in_sort,
+    ],
+    logy=True,
+    labels=[
+        "bubble sort",
+        "insertion sort",
+        "merge sort",
+        "quick sort",
+        "tim sort",
+        "built in sort",
+    ],
+    xlabel="Number of elements in list",
+)

python-perfplot/plot_sorting_powers_of_two.py

@@ -0,0 +1,25 @@
+from random import randint
+
+import perfplot
+
+from sorting import (
+    bubble_sort,
+    insertion_sort,
+    merge_sort,
+    quick_sort,
+    tim_sort,
+    python_built_in_sort,
+)
+
+perfplot.show(
+    n_range=[2**n for n in range(15)],
+    setup=lambda n: [randint(0, 1000) for _ in range(n)],
+    kernels=[
+        bubble_sort,
+        insertion_sort,
+        merge_sort,
+        quick_sort,
+        tim_sort,
+        python_built_in_sort,
+    ],
+)

python-perfplot/plot_sorting_pretty.py

@@ -0,0 +1,37 @@
+from random import randint
+
+import perfplot
+
+from sorting import (
+    bubble_sort,
+    insertion_sort,
+    merge_sort,
+    quick_sort,
+    tim_sort,
+    python_built_in_sort,
+)
+
+perfplot.show(
+    n_range=[2**n for n in range(15)],
+    setup=lambda n: [randint(0, 1_000) for _ in range(n)],
+    kernels=[
+        bubble_sort,
+        insertion_sort,
+        merge_sort,
+        quick_sort,
+        tim_sort,
+        python_built_in_sort,
+    ],
+    logy=True,
+    labels=[
+        "bubble sort",
+        "insertion sort",
+        "merge sort",
+        "quick sort",
+        "tim sort",
+        "built in sort",
+    ],
+    title="Sorting Algorithms",
+    xlabel="Number of elements in list",
+    time_unit="ns",
+)