Attempt module: make traits used in tests only conditional

elabit · Sep 18, 2023 · f4414b7 · f4414b7
1 parent 57a1a00
commit f4414b7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/v2/rust/src/attempt.rs b/v2/rust/src/attempt.rs
@@ -3,7 +3,8 @@ use std::process::Command;
 
 const PYTHON_EXECUTABLE: &str = "python";
 
-#[derive(Clone, PartialEq, Debug)]
+#[derive(Clone, PartialEq)]
+#[cfg_attr(test, derive(Debug))]
 enum RetryStrategy {
     Incremental,
     Complete,
@@ -15,7 +16,8 @@ struct Variant {
     argument_file: Option<PathBuf>,
 }
 
-#[derive(Clone, PartialEq, Debug)]
+#[derive(Clone)]
+#[cfg_attr(test, derive(Debug, PartialEq))]
 pub struct Identifier {
     name: String,
     timestamp: String,
@@ -38,7 +40,7 @@ impl RetrySpec {
     }
 }
 
-#[derive(PartialEq, Debug)]
+#[cfg_attr(test, derive(Debug, PartialEq))]
 pub struct Attempt {
     output_directory: PathBuf,
     identifier: Identifier,