feat(t-wise): test for the correct number of interactions generated

Author: uulm-janbaudisch

ddnnife/src/ddnnf/anomalies/t_wise_sampling.rs

@@ -146,8 +146,8 @@ impl ExtendedDdnnf {
 
 #[cfg(test)]
 mod test {
-    use crate::ddnnf::anomalies::t_wise_sampling::Sample;
     use crate::ddnnf::anomalies::t_wise_sampling::t_iterator::TInteractionIter;
+    use crate::ddnnf::anomalies::t_wise_sampling::{Sample, SamplingResult};
     use crate::ddnnf::extended_ddnnf::optimal_configs::test::build_sandwich_ext_ddnnf_with_objective_function_values;
     use crate::{Ddnnf, parser::build_ddnnf};
     use itertools::Itertools;
@@ -231,4 +231,18 @@ mod test {
             check_validity_of_sample(&ext_ddnnf.sample_t_wise_yasa(t), &ext_ddnnf.ddnnf, t);
         }
     }
+
+    #[test]
+    fn number_of_interactions() {
+        let ddnnf = build_ddnnf(Path::new("tests/data/busybox.nnf"), Some(42));
+        let sample = ddnnf.sample_t_wise(2);
+        match sample {
+            SamplingResult::Empty | SamplingResult::Void => {
+                panic!("Expected valid, non-empty sampling result.")
+            }
+            SamplingResult::ResultWithSample(sample) => {
+                assert_eq!(sample.interactions(2).len(), 1387048)
+            }
+        }
+    }
 }

ddnnife/src/ddnnf/anomalies/t_wise_sampling/config.rs

@@ -1,6 +1,8 @@
 use super::SatWrapper;
+use crate::ddnnf::anomalies::t_wise_sampling::t_iterator::TInteractionIter;
 use std::fmt::Display;
 use std::hash::{Hash, Hasher};
+use streaming_iterator::StreamingIterator;
 
 /// Represents a (partial) configuration
 #[derive(Debug, Clone, Eq)]
@@ -203,4 +205,13 @@ impl Config {
     pub fn is_complete(&self) -> bool {
         self.n_decided_literals == self.literals.len()
     }
+
+    /// Generates all `t`-wise interactions covered by this configuration.
+    pub fn interactions(&self, t: usize) -> Vec<Vec<i32>> {
+        let mut out = Vec::new();
+        TInteractionIter::new(&self.literals, t)
+            .for_each(|interaction| out.push(interaction.to_vec()));
+
+        out
+    }
 }

ddnnife/src/ddnnf/anomalies/t_wise_sampling/sample.rs

@@ -236,6 +236,22 @@ impl Sample {
         TInteractionIter::new(&literals, min(t, literals.len()))
             .all(|interaction| self.covers(interaction))
     }
+
+    /// Generates all `t`-wise interactions covered by this sample.
+    pub fn interactions(&self, t: usize) -> HashSet<Vec<i32>> {
+        let mut interactions = HashSet::new();
+        self.iter().for_each(|config| {
+            config
+                .interactions(t)
+                .into_iter()
+                .for_each(|mut interaction| {
+                    interaction.sort_unstable();
+                    interactions.insert(interaction);
+                });
+        });
+
+        interactions
+    }
 }
 
 #[cfg(test)]