Refactored CIM validation for `UniformParametersGenerator` test

pull/85/head
Meliurwen 2 years ago
parent f4e3c98c79
commit 0f61cbee4c
Signed by: meliurwen
GPG Key ID: 818A8B35E9F1CE10
  1. 8
      reCTBN/tests/tools.rs

@ -1,6 +1,7 @@
use std::ops::Range; use std::ops::Range;
use ndarray::{arr1, arr2, arr3}; use ndarray::{arr1, arr2, arr3};
use reCTBN::params::ParamsTrait;
use reCTBN::process::ctbn::*; use reCTBN::process::ctbn::*;
use reCTBN::process::NetworkProcess; use reCTBN::process::NetworkProcess;
use reCTBN::params; use reCTBN::params;
@ -167,12 +168,9 @@ fn uniform_parameters_generator_right_densities() {
let mut cim_generator: UniformParametersGenerator = RandomParametersGenerator::new(interval, seed); let mut cim_generator: UniformParametersGenerator = RandomParametersGenerator::new(interval, seed);
cim_generator.generate_parameters(&mut net); cim_generator.generate_parameters(&mut net);
for node in net.get_node_indices() { for node in net.get_node_indices() {
match &mut net.get_node_mut(node) {
params::Params::DiscreteStatesContinousTime(param) => {
assert_eq!( assert_eq!(
Ok(()), Ok(()),
param.set_cim(param.get_cim().clone().unwrap())); net.get_node(node).validate_params()
} );
}
} }
} }

Loading…
Cancel
Save