Rust中能否仅在测试中启用针对内部不变量的断言？

可以通过Rust的条件编译（cfg(test)）来实现仅在测试中启用内部不变量的断言，同时避免非测试代码的性能开销。以下是具体实现方式：

1. 限定不变量检测函数仅在测试中编译 #

给你的test_invariant函数加上#[cfg(test)]属性，这样非测试编译时会完全忽略这个函数，避免不必要的编译和运行开销：

#[cfg(test)]
fn test_invariant(&self) -> bool { 
    // 你的不变量检测逻辑
}

2. 自定义测试专用断言宏 #

创建一个仅在测试模式下生效的断言宏，非测试模式下会被展开为空语句：

#[cfg(test)]
macro_rules! test_invariant_assert {
    ($instance:expr) => {
        debug_assert!($instance.test_invariant(), "内部不变量被违反");
    };
}

#[cfg(not(test))]
macro_rules! test_invariant_assert {
    ($instance:expr) => {};
}

3. 在算法中使用自定义断言 #

在你的算法流程中，用这个自定义宏替代原来的debug_assert!：

fn some_algorithm(&self) {
    // 算法前半部分代码

    test_invariant_assert!(self);

    // 算法后半部分代码
}

原理说明 #

• cfg(test)会在执行cargo test时自动启用，此时断言和检测函数都会被编译并执行；
• 当用户编译你的库/程序（cargo build或cargo run）时，cfg(test)不生效，检测函数不会被编译，断言宏也不会生成任何代码，完全没有性能损耗；
• 这种方式完美适配你提到的「原地算法中无法将不变量检测移到单元测试」的场景，因为断言会嵌入到算法的关键执行节点。

另外，如果需要在测试的release模式下也启用断言（默认cargo test用debug模式），可以通过cargo test --release执行，此时cfg(test)依然生效，断言会正常触发。

内容的提问来源于stack exchange，提问作者Nikita Gaevoy