schrodinger.seam.testing.benchmarks module

Usage:

$SCHRODINGER/run python3 -m schrodinger.seam.testing.benchmarks

class schrodinger.seam.testing.benchmarks.BenchmarkResult(time_in_seconds: float)

Bases: object

time_in_seconds: float

__init__(time_in_seconds: float) → None

class schrodinger.seam.testing.benchmarks.Benchmark

Bases: object

setup()

run(n: int)

class schrodinger.seam.testing.benchmarks.GroupByKeyWithSlowCoder

Bases: schrodinger.seam.testing.benchmarks.Benchmark

run(n: int) → schrodinger.seam.testing.benchmarks.BenchmarkResult

This benchmark is to test the performance of GroupByKey when using keys and values that are slow to deserialize.

class schrodinger.seam.testing.benchmarks.ExecStagesWithSlowCoder

Bases: schrodinger.seam.testing.benchmarks.Benchmark

run(n: int) → schrodinger.seam.testing.benchmarks.BenchmarkResult

This benchmark is to test the performance of ExecutableStages which have outputs that are slow to deserialize.

class schrodinger.seam.testing.benchmarks.FlattenWithSlowCoder

Bases: schrodinger.seam.testing.benchmarks.Benchmark

run(n: int) → schrodinger.seam.testing.benchmarks.BenchmarkResult

This benchmark is to test the performance of Flatten when using elements that are slow to deserialize.

class schrodinger.seam.testing.benchmarks.GroupByKey

Bases: schrodinger.seam.testing.benchmarks.Benchmark

property gbk_executor

Set up a GroupByKeyExecutor with an input and output EmbeddedPCollManager

Groups input type of (int, int) to output type of (int, Iterable[int])

run(n: int) → schrodinger.seam.testing.benchmarks.BenchmarkResult

schrodinger.seam.testing.benchmarks.get_benchmarks()

schrodinger.seam.testing.benchmarks.main()