More test passing for History

src/main.rs

@@ -3,37 +3,38 @@ use std::collections::HashMap;
 use trueskill_tt::*;
 
 fn main() {
-    let mut agents = HashMap::new();
+    let composition = vec![
+        vec![vec!["aj"], vec!["bj"]],
+        vec![vec!["bj"], vec!["cj"]],
+        vec![vec!["cj"], vec!["aj"]],
+    ];
+    let results = vec![vec![1, 0], vec![1, 0], vec![1, 0]];
+    let times = vec![1, 2, 3];
 
-    for k in ["a", "b", "c", "d", "e", "f"] {
-        let agent = Agent::new(
-            Player::new(
-                Gaussian::new(25.0, 25.0 / 3.0),
-                25.0 / 6.0,
-                25.0 / 300.0,
-                N_INF,
-            ),
+    let mut priors = HashMap::new();
+
+    for k in ["aj", "bj", "cj"] {
+        let player = Player::new(
+            Gaussian::new(25.0, 25.0 / 3.0),
+            25.0 / 6.0,
+            25.0 / 300.0,
             N_INF,
-            f64::NEG_INFINITY,
         );
 
-        agents.insert(k.to_string(), agent);
+        priors.insert(k.to_string(), player);
     }
 
-    let b = Batch::new(
-        vec![
-            vec![vec!["a"], vec!["b"]],
-            vec![vec!["c"], vec!["d"]],
-            vec![vec!["e"], vec!["f"]],
-        ],
-        vec![vec![1, 0], vec![0, 1], vec![1, 0]],
-        0.0,
-        agents,
-        0.0,
+    let mut h2 = History::new(
+        composition,
+        results,
+        times,
+        priors,
+        MU,
+        BETA,
+        SIGMA,
+        GAMMA,
+        P_DRAW,
     );
 
-    let post = b.posteriors();
-
-    println!("{} {}", post["a"].mu(), 29.205);
-    println!("{} {}", post["a"].sigma(), 7.194)
+    let (step, i) = h2.convergence();
 }