diff --git a/board/src/position/flags.rs b/board/src/position/flags.rs index 2aa389d..7efc41d 100644 --- a/board/src/position/flags.rs +++ b/board/src/position/flags.rs @@ -69,5 +69,11 @@ mod tests { assert!(!flags.player_has_right_to_castle(Color::White, BoardSide::Queen)); assert!(flags.player_has_right_to_castle(Color::Black, BoardSide::King)); assert!(flags.player_has_right_to_castle(Color::Black, BoardSide::Queen)); + + flags.set_player_has_right_to_castle_flag(Color::White, BoardSide::Queen); + assert!(flags.player_has_right_to_castle(Color::White, BoardSide::King)); + assert!(flags.player_has_right_to_castle(Color::White, BoardSide::Queen)); + assert!(flags.player_has_right_to_castle(Color::Black, BoardSide::King)); + assert!(flags.player_has_right_to_castle(Color::Black, BoardSide::Queen)); } }