Update king move generator tests

This commit is contained in:
Eryn Wells 2024-01-06 16:40:21 -08:00
parent 216140bdbc
commit 3b266cb94a

View file

@ -161,13 +161,13 @@ impl<'pos> Iterator for KingMoveGenerator<'pos> {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
use crate::{Position, Square}; use crate::{piece::Piece, Position, Square};
use std::collections::HashSet; use std::collections::HashSet;
#[test] #[test]
fn one_king() { fn one_king() {
let mut pos = Position::empty(); let mut pos = Position::empty();
pos.place_piece(&Piece::king(Color::White), &Square::e4()) pos.place_piece(Piece::king(Color::White), Square::E4)
.expect("Failed to place king on e4"); .expect("Failed to place king on e4");
let generator = KingMoveGenerator::new(&pos, Color::White); let generator = KingMoveGenerator::new(&pos, Color::White);
@ -211,7 +211,7 @@ mod tests {
#[test] #[test]
fn one_king_corner() { fn one_king_corner() {
let mut pos = Position::empty(); let mut pos = Position::empty();
pos.place_piece(&Piece::king(Color::White), &Square::a1()) pos.place_piece(Piece::king(Color::White), Square::A1)
.expect("Failed to place king on a1"); .expect("Failed to place king on a1");
let generator = KingMoveGenerator::new(&pos, Color::White); let generator = KingMoveGenerator::new(&pos, Color::White);