Mark method which is used in tests only

Cardinal-Cryptography · Jan 13, 2025 · 3c8af55 · 3c8af55
1 parent 8078477
commit 3c8af55
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/consensus/src/dag/reconstruction/mod.rs b/consensus/src/dag/reconstruction/mod.rs
@@ -96,7 +96,8 @@ impl<U: Unit> UnitWithParents for ReconstructedUnit<U> {
     fn parent_for(&self, index: NodeIndex) -> Option<&HashFor<Self>> {
         self.parents.get(index).map(|(hash, _)| hash)
     }
-
+
+    #[cfg(test)]
     fn parent_round(&self, index: NodeIndex) -> Option<Round> {
         self.parents.get(index).map(|(_, round)| *round)
     }

diff --git a/consensus/src/units/mod.rs b/consensus/src/units/mod.rs
@@ -226,6 +226,7 @@ pub trait UnitWithParents: Unit {
     fn direct_parents(&self) -> impl Iterator<Item = &HashFor<Self>>;
     fn parent_for(&self, index: NodeIndex) -> Option<&HashFor<Self>>;
 
+    #[cfg(test)]
     fn parent_round(&self, index: NodeIndex) -> Option<Round>;
     fn node_count(&self) -> NodeCount;
 }