Skip to main content

A simple and easy-to-use Python benchmarking library

Project description

EasyBench

Tests Docs PyPI version License: MIT Python 3.10+ Checked with mypy Black Ruff

Docs

A simple and easy-to-use Python benchmarking library.

Features

  • Three benchmarking styles: decorator, class-based, and command-line
  • Measure both execution time and memory usage (see limitations)
  • Rich visualizations (Boxplot, Violinplot, Lineplot, Histogram, Barplot)
  • Advanced options: warmup runs, multiple loops per trial, outlier trimming
  • Parametrized benchmarks for comparing function performance with different inputs
  • pytest-like fixtures and lifecycle hooks (setup/teardown)
  • Flexible configuration: time/memory units, progress tracking, filtering
  • Multiple output formats (text tables, CSV, JSON, pandas.DataFrame)
  • Extensible reporting system for custom outputs

Installation

pip install easybench

Optional Dependencies

EasyBench supports optional dependencies for additional features:

# Install with visualization support
pip install easybench[all]

The all option includes:

  • matplotlib: For visualization and plotting benchmark results
  • seaborn: For enhanced statistical visualizations
  • pandas: For outputting benchmark results as DataFrames
  • tqdm: For progress tracking during benchmark execution

Quick Start

Open In Colab

There are 3 ways to benchmark with easybench:

  1. @bench decorator

    from easybench import bench
    
    # Add @bench with function parameters
    @bench(item=123, big_list=lambda: list(range(1_000_000)))
    def insert_first(item, big_list):
        big_list.insert(0, item)
    
    • When you need fresh data for each trial, use a function or lambda to generate new data on demand.
      (like lambda: list(range(1_000_000)) in the above)
  2. EasyBench class

    from easybench import EasyBench, BenchConfig
    
    class BenchListOperation(EasyBench):
        # Benchmark configuration
        bench_config = BenchConfig(
            trials=10,
            memory=True,
            sort_by="avg"
        )
    
        # Setup for each trial
        def setup_trial(self):
            self.big_list = list(range(1_000_000))
    
        # Benchmark methods (must start with bench_)
        def bench_insert_first(self):
            self.big_list.insert(0, 123)
    
        # You can define multiple benchmark methods
        def bench_pop_first(self):
            self.big_list.pop(0)
    
    if __name__ == "__main__":
        # Run benchmark
        BenchListOperation().bench()
    
  3. easybench command

    1. Create a benchmarks directory

    2. Put bench_*.py scripts in the directory:

      from easybench import fixture
      
      # Fixture for each trial
      @fixture(scope="trial")
      def big_list():
          return list(range(1_000_000))
      
      # Benchmark functions (must start with bench_)
      def bench_insert_first(big_list):
          big_list.insert(0, 123)
      
      # You can define multiple benchmark functions
      def bench_pop_first(big_list):
          big_list.pop(0)
      
    3. Run easybench command

      easybench --trials 10 --memory --sort-by avg
      

Example of benchmark results:

  • Single benchmark

    Benchmark Results (5 trials):
    
    Function        Avg Time (s)  Min Time (s)  Max Time (s)
    --------------------------------------------------------
    insert_first        0.001568      0.001071      0.003265
    
  • Multiple benchmarks

    EasyBench Benchmark Result

  • Boxplot Visualization

    Boxplot Visualization

  • Violinplot Visualization

    Violinplot Visualization

  • Lineplot Visualization

    Lineplot Visualization

  • Histogram Visualization

    Histplot Visualization

  • Barplot Visualization

    Barplot Visualization

Usage

For detailed usage instructions, please refer to the documentation.

Memory Measurement Limitations

EasyBench uses Python's built-in tracemalloc module to measure memory usage.
This has some important limitations:

  • tracemalloc only tracks memory allocations made through Python's memory manager
  • Memory allocated by C extensions (like NumPy, Pandas, or other native libraries) often bypasses Python's memory manager and won't be accurately measured
  • The reported memory usage reflects Python objects only, not the total process memory consumption For applications heavily using C extensions, consider using external profilers like memory_profiler or system monitoring tools for more accurate measurements.

License

MIT

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

easybench-0.1.32.tar.gz (445.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

easybench-0.1.32-py3-none-any.whl (41.5 kB view details)

Uploaded Python 3

File details

Details for the file easybench-0.1.32.tar.gz.

File metadata

  • Download URL: easybench-0.1.32.tar.gz
  • Upload date:
  • Size: 445.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.10.18

File hashes

Hashes for easybench-0.1.32.tar.gz
Algorithm Hash digest
SHA256 4378043d316d73f072eda76a4302377a8dc1982a7ea5afaacf4cf72a76a281a6
MD5 b738ae680c54739eb9dcc3a19c672f68
BLAKE2b-256 b66a2deb8cf13e58b79f6a2e53a41bb33d8f77b5ee49a9abc6c12cb3d552106e

See more details on using hashes here.

File details

Details for the file easybench-0.1.32-py3-none-any.whl.

File metadata

  • Download URL: easybench-0.1.32-py3-none-any.whl
  • Upload date:
  • Size: 41.5 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.10.18

File hashes

Hashes for easybench-0.1.32-py3-none-any.whl
Algorithm Hash digest
SHA256 313fa3b14eddd608641435ce9ffd1e23a12251e7e962b41fccb23962ce6885ce
MD5 2cdf1dea171dcb9a28a29af324e5069b
BLAKE2b-256 5400ebf1ee3c4c57cbbf40c384142414820ef3f69717d86526e38f62a4538a93

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page