2

我开始在 Rust 中使用“bench”和 Rust nightly 编写基准测试,如文档中所写。

为了在测试和基准测试之间共享代码,我添加Option<&mut Bencher>了直接运行代码块或通过 bencher 运行代码块(“src/lib.rs”):

fn block_requests(bencher_option: Option<&mut Bencher>, ...) {
    ...

    let mut block = || {
       ... // shared
    }

    match bencher_option {
            // regular test
            None => block(),

            // benchmark
            Some(bencher) => {
                bencher.iter(block);
            }
        }
    ...
}

// call from test
#[test]
fn test_smth() {
    block_requests(None, &requests, &mut matcher);
}


// call from benchmark
#[bench]
fn bench_smth(b: &mut Bencher) {
    block_requests(Some(b), &requests, &mut matcher);
}

现在我想使用 Rust stable进行基准测试。由于“bencher”板条箱3 年没有更新, “criterion”板条箱似乎是默认选项。为此,我必须将代码移动到“./benches/my_benchmark.rs”。

我怎样才能block_requests(..)在测试和基准测试之间共享?

4

1 回答 1

0

tests/并且benches/工作方式与src/main.rsor相同src/bin/*.rs:它们是单独的二进制 crate。这意味着它们必须按名称引用库 crate 中的项目,并且这些项目必须是可见的。

所以,你需要改变

fn block_requests(bencher_option: Option<&mut Bencher>, ...) {

使其成为public,您可能还想添加#[doc(hidden)],以便您的库文档不包含测试助手:

#[doc(hidden)]
pub fn block_requests(bencher_option: Option<&mut Bencher>, ...) {

然后,在您的测试和基准测试中,use通过给出您的 crate 的名称。

use my_library_crate_name::block_requests;

fn bench_smth(...) {...}

(您不能use crate::block_requests在此处使用,因为关键字crate指的是基准二进制包,而不是库包。)

于 2021-02-11T15:50:46.153 回答