From 461ed5b388dd7fdf30b84cfb56db7e467fad0cd6 Mon Sep 17 00:00:00 2001
From: Sebastian <sebastian.schirmer@dlr.de>
Date: Thu, 6 Jun 2024 16:15:09 +0200
Subject: [PATCH] fixed tests

---
 scripts/run.sh |  4 ++--
 tests/tests.rs | 23 +++++++++++++----------
 2 files changed, 15 insertions(+), 12 deletions(-)

diff --git a/scripts/run.sh b/scripts/run.sh
index 05cb712..8fde642 100644
--- a/scripts/run.sh
+++ b/scripts/run.sh
@@ -25,10 +25,10 @@ directories=$(find "$script_dir/../res/" -mindepth 1 -type d -printf "%P\n")
 for dir in $directories; do
     dir="../res/${dir}/"
     echo "Running Lazy eval with subsampling using ${dir}"
-    ../target/release/tbt-segmentation -l -c -u -s ../specification/shiplanding_formula_combined.tbt -f $dir> "${dir}/subsamplingAndLazy_result.txt"
+    ../target/release/tbt-segmentation -l -c -u -s ../specification/shiplanding_formula_combined.tbt --toml "${dir}/subsamplingAndLazy_result.toml" -f $dir> "${dir}/subsamplingAndLazy_result.txt"
     python3 infer_parameters_visualization.py "${dir}/subsamplingAndLazy_result.txt"
     echo "Running eval with subsampling using ${dir}"
-    ../target/release/tbt-segmentation -c -u -s ../specification/shiplanding_formula_combined.tbt -f $dir> "${dir}/subsampling_result.txt"
+    ../target/release/tbt-segmentation -c -u -s ../specification/shiplanding_formula_combined.tbt --toml "${dir}/subsampling_result.toml" -f $dir> "${dir}/subsampling_result.txt"
     python3 infer_parameters_visualization.py "${dir}/subsampling_result.txt"
 done
 
diff --git a/tests/tests.rs b/tests/tests.rs
index 7451b61..6e9de78 100644
--- a/tests/tests.rs
+++ b/tests/tests.rs
@@ -43,7 +43,7 @@ fn run_test(
 ) -> Result<(), String> {
     for (trace, expected) in traces_with_expected_value {
         let trace: Trace = (trace.len(), HashMap::from([(signal_name.clone(), trace)]));
-        let robustness = evaluate(
+        let ((robustness, _), _) = evaluate(
             tbt.clone(),
             trace,
             SystemTime::now(),
@@ -54,10 +54,10 @@ fn run_test(
             None,
             false,
         );
-        if robustness.0 == expected {
+        if robustness == expected {
             continue;
         } else {
-            return Err(format!("Expected {expected} but was {}.", robustness.0));
+            return Err(format!("Expected {expected} but was {}.", robustness));
         }
     }
     Ok(())
@@ -87,7 +87,7 @@ fn test_hardcoded_maneuver(specification: &str, logfile: &str) {
     let setting_2 = setting.clone();
     let start = SystemTime::now();
     println!("Run using parsed tbt!");
-    let first_run = evaluate(
+    let (first_run, first_run_alt) = evaluate(
         new_tbt,
         trace,
         start,
@@ -99,7 +99,7 @@ fn test_hardcoded_maneuver(specification: &str, logfile: &str) {
         false,
     );
     println!("Run using hand-coded tbt!");
-    let second_run = evaluate(
+    let (second_run, second_run_alt) = evaluate(
         old_tbt,
         trace_2,
         start,
@@ -140,12 +140,15 @@ fn test_hardcoded_maneuver(specification: &str, logfile: &str) {
             v2.3
         );
     }
-    let alternatives_new = first_run.2.unwrap();
-    let alternatives_old = second_run.2.unwrap();
+    let alternatives_new = first_run_alt.unwrap();
+    let alternatives_old = second_run_alt.unwrap();
     for j in 0..alternatives_new.len() {
-        for i in 0..first_run.1.len() {
-            let v1 = &alternatives_new[j][i];
-            let v2 = &alternatives_old[j][i];
+        let alt_new = &alternatives_new[j];
+        let alt_old = &alternatives_old[j];
+        assert_eq!(alt_new.0, alt_old.0);
+        for i in 0..alt_old.1.len() {
+            let v1 = &alt_new.1[i];
+            let v2 = &alt_old.1[i];
             assert!(
                 v1.1 == v2.1,
                 "Entry of alternatives i={i} j={j} is expected to have same lower but wasnt {} != {}",