Use named function in incremental red knot benchmark (#14033)

This commit is contained in:
Micha Reiser 2024-11-01 09:44:38 +01:00 committed by GitHub
parent cf0f5e1318
commit 48fa839c80
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -120,13 +120,11 @@ fn setup_rayon() {
} }
fn benchmark_incremental(criterion: &mut Criterion) { fn benchmark_incremental(criterion: &mut Criterion) {
setup_rayon(); fn setup() -> Case {
criterion.bench_function("red_knot_check_file[incremental]", |b| {
b.iter_batched_ref(
|| {
let case = setup_case(); let case = setup_case();
case.db.check().unwrap(); let result = case.db.check().unwrap();
assert_eq!(result, EXPECTED_DIAGNOSTICS);
case.fs case.fs
.write_file( .write_file(
@ -136,8 +134,9 @@ fn benchmark_incremental(criterion: &mut Criterion) {
.unwrap(); .unwrap();
case case
}, }
|case| {
fn incremental(case: &mut Case) {
let Case { db, .. } = case; let Case { db, .. } = case;
db.apply_changes( db.apply_changes(
@ -151,9 +150,12 @@ fn benchmark_incremental(criterion: &mut Criterion) {
let result = db.check().unwrap(); let result = db.check().unwrap();
assert_eq!(result, EXPECTED_DIAGNOSTICS); assert_eq!(result, EXPECTED_DIAGNOSTICS);
}, }
BatchSize::SmallInput,
); setup_rayon();
criterion.bench_function("red_knot_check_file[incremental]", |b| {
b.iter_batched_ref(setup, incremental, BatchSize::SmallInput);
}); });
} }