diff --git a/src/game.rs b/src/game.rs index 85d61a2..1cd799e 100644 --- a/src/game.rs +++ b/src/game.rs @@ -229,7 +229,7 @@ impl Game { mod tests { use ::approx::assert_ulps_eq; - use crate::{Gaussian, Player, GAMMA}; + use crate::{Gaussian, Player, GAMMA, N_INF}; use super::*; @@ -264,9 +264,8 @@ mod tests { let g = Game::new(vec![vec![t_a], vec![t_b]], vec![0.0, 1.0], vec![], 0.0); - assert_eq!(g.likelihoods[0][0].sigma, f64::INFINITY); - assert_eq!(g.likelihoods[1][0].sigma, f64::INFINITY); - assert_eq!(g.likelihoods[0][0].mu, 0.0); + assert_eq!(g.likelihoods[0][0], N_INF); + assert_eq!(g.likelihoods[1][0], N_INF); } #[test]