Add rook movement tests and refactor rook movement type search