wordle-analyzer/tests/solver.rs

use test_log::test; // set the log level with an envvar: `RUST_LOG=trace cargo test`

use wordle_analyzer::game::evaluation::Evaluation;
use wordle_analyzer::game::Game;
use wordle_analyzer::solve::{AnyBuiltinSolver, NaiveSolver, Solver, StupidSolver};
use wordle_analyzer::wlist::builtin::BuiltinWList;
use wordle_analyzer::wlist::word::{Word, WordData};
use wordle_analyzer::wlist::WordList;

fn wordlist() -> impl WordList {
    BuiltinWList::default()
}

#[test]
fn test_build_builtin_solvers() {
    let wl = wordlist();
    let _stupid_solver =
        AnyBuiltinSolver::Stupid(StupidSolver::build(&wl).expect("could not build naive solver"));
    let _naive_solver =
        AnyBuiltinSolver::Naive(NaiveSolver::build(&wl).expect("could not build naive solver"));
}

#[test]
fn test_naive_play_predetermined_game() -> anyhow::Result<()> {
    let wl = wordlist();
    let sl =
        AnyBuiltinSolver::Naive(NaiveSolver::build(&wl).expect("could not build naive solver"));
    let mut game = Game::build(5, false, 6, &wl, false)?;
    game.set_solution(Some(("nines".into(), 0.002))); // The accuracy is made up but shouldn't
                                                      // matter
    sl.make_a_move(&mut game)?;
    assert_eq!(
        game.responses().last().unwrap().guess(),
        &Word::from("which")
    );
    sl.make_a_move(&mut game)?;
    assert_eq!(
        game.responses().last().unwrap().guess(),
        &Word::from("their")
    );
    sl.make_a_move(&mut game)?;
    assert_eq!(
        game.responses().last().unwrap().guess(),
        &Word::from("being")
    );
    sl.make_a_move(&mut game)?;
    assert_eq!(
        game.responses().last().unwrap().guess(),
        &Word::from("since")
    );
    sl.make_a_move(&mut game)?;
    assert_eq!(
        game.responses().last().unwrap().guess(),
        &Word::from("lines")
    );
    sl.make_a_move(&mut game)?;
    assert_eq!(
        game.responses().last().unwrap().guess(),
        &Word::from("mines")
    );
    sl.make_a_move(&mut game)?;
    assert_eq!(
        game.responses().last().unwrap().guess(),
        &Word::from("wines")
    );

    // naive is at the moment too bad to solve "nines"
    assert!(game.finished());
    assert!(!game.won());

    Ok(())
}

#[test]
fn test_naive_play_predetermined_game_manually() -> anyhow::Result<()> {
    let wl = wordlist();
    let sl =
        AnyBuiltinSolver::Naive(NaiveSolver::build(&wl).expect("could not build naive solver"));
    // we don't insert the solution yet,
    // pretend that a user inputs guesses manually
    let mut game = Game::build(5, false, 6, &wl, false)?;
    let _actual_solution: Option<WordData> = Some(("nines".into(), 0.002));
    let mut next_guess: Word;

    next_guess = sl.guess_for(&game)?;
    assert_eq!(next_guess, Word::from("which"));
    game.guess(
        &next_guess.clone(),
        Some(Evaluation::build(&next_guess, "xxfxx")?),
    )?;

    next_guess = sl.guess_for(&game)?;
    assert_eq!(next_guess, Word::from("their"));
    game.guess(
        &next_guess.clone(),
        Some(Evaluation::build(&next_guess, "xxffx")?),
    )?;

    next_guess = sl.guess_for(&game)?;
    assert_eq!(next_guess, Word::from("being"));
    game.guess(
        &next_guess.clone(),
        Some(Evaluation::build(&next_guess, "xfffx")?),
    )?;

    next_guess = sl.guess_for(&game)?;
    assert_eq!(next_guess, Word::from("since"));
    game.guess(
        &next_guess.clone(),
        Some(Evaluation::build(&next_guess, "fcfxf")?),
    )?;

    next_guess = sl.guess_for(&game)?;
    assert_eq!(next_guess, Word::from("lines"));
    game.guess(
        &next_guess.clone(),
        Some(Evaluation::build(&next_guess, "xcccc")?),
    )?;

    next_guess = sl.guess_for(&game)?;
    assert_eq!(next_guess, Word::from("mines"));
    game.guess(
        &next_guess.clone(),
        Some(Evaluation::build(&next_guess, "xcccc")?),
    )?;

    next_guess = sl.guess_for(&game)?;
    assert_eq!(next_guess, Word::from("wines"));
    game.guess(
        &next_guess.clone(),
        Some(Evaluation::build(&next_guess, "xcccc")?),
    )?;

    // naive is at the moment too bad to solve "nines"
    assert!(game.finished());
    assert!(!game.won());

    Ok(())
}