Add test for rook moves