Updated test utils to enhance error handling for non-deterministic effects; added unit test for erroneous singleton evaluation.

Signed-off-by: Marvin Hansen <[email protected]>

Author: marvin-hansen

deep_causality/src/utils_test/test_utils.rs

@@ -144,16 +144,15 @@ pub fn get_test_causaloid_deterministic_input_output() -> BaseCausaloid {
     let description = "Inverts any input";
 
     fn causal_fn(effect: &PropagatingEffect) -> Result<PropagatingEffect, CausalityError> {
-        let obs =
-            match effect {
-                // If it's the Deterministic variant, extract the inner value.
-                PropagatingEffect::Deterministic(val) => *val,
-                // For any other type of effect, this function cannot proceed, so return an error.
-                _ => return Err(CausalityError(
-                    "Causal function expected Numerical effect but received a different variant."
-                        .into(),
-                )),
-            };
+        let obs = match effect {
+            // If it's the Deterministic variant, extract the inner value.
+            PropagatingEffect::Deterministic(val) => *val,
+            // For any other type of effect, this function cannot proceed, so return an error.
+            _ => return Err(CausalityError(
+                "Causal function expected Deterministic effect but received a different variant."
+                    .into(),
+            )),
+        };
 
         // Just invert the value.
         Ok(PropagatingEffect::Deterministic(!obs))

deep_causality/tests/types/causal_types/causaloid/causaloid_tests.rs

@@ -6,7 +6,9 @@
 use deep_causality::*;
 use std::sync::Arc;
 
-use deep_causality::utils_test::test_utils::get_base_context;
+use deep_causality::utils_test::test_utils::{
+    get_base_context, get_test_causaloid_deterministic_input_output,
+};
 use deep_causality::utils_test::*;
 
 // Helper function to unpack numerical evidence, used in test causal functions.
@@ -415,6 +417,17 @@ fn test_evaluate_singleton_with_context() {
     assert_eq!(res_false, PropagatingEffect::Deterministic(false));
 }
 
+#[test]
+fn test_evaluate_singleton_err() {
+    let causaloid: BaseCausaloid = get_test_causaloid_deterministic_input_output();
+
+    // The causal function expects a Deterministic effect, but we pass in a Probabilistic effect.
+    let effect = PropagatingEffect::Probabilistic(4.2);
+    // The result should be an error.
+    let res = causaloid.evaluate(&effect);
+    assert!(res.is_err());
+}
+
 #[test]
 fn test_evaluate_collection_other_effect_err() {
     // Setup: A collection with causaloids that have effects other than Deterministic or Halting.