dogeystamp/chess_inator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: fix broken nnue binary test

Browse Source
This commit is contained in:
dogeystamp 2024-12-28 22:28:50 -05:00
parent 47a600cd80
commit 87501b5c94
No known key found for this signature in database
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/nnue.rs
View File

@ -80,12 +80,13 @@ impl Display for InputTensor {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
#[test]
fn test_to_binary_tensor() { fn test_to_binary_tensor() {
// more of a sanity check than a test // more of a sanity check than a test
let board = Board::from_fen("8/8/8/8/8/8/8/1b6 w - - 0 1").unwrap(); let board = Board::from_fen("8/8/8/8/8/8/8/1b6 w - - 0 1").unwrap();
let tensor = InputTensor::from_board(&board); let tensor = InputTensor::from_board(&board);
let mut expected = [false; INP_TENSOR_SIZE]; let mut expected = [false; INP_TENSOR_SIZE];
expected[1 + 1 + INP_TENSOR_SIZE / 2] = true; expected[INP_TENSOR_SIZE / N_COLORS + 1 + N_SQUARES] = true;
assert_eq!(tensor.0, expected); assert_eq!(tensor.0, expected);
} }
} }