[Chess] Enhance CAN_MOVE

[sotetsuk]

#1232 #1233

fn	jaxpr lines	time
_legal_action_mask	1674	427.0ms
_flip	92	28.4ms
_is_checked	203	52.8ms
_is_pseudo_legal	134	35.8ms
_hash_castling_en_passant	37	17.6ms
_apply_move	331	92.8ms
_update_history	57	27.5ms

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 80.63%. Comparing base (b9ad95a) to head (53cb468).
Report is 1 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1234      +/-   ##
==========================================
+ Coverage   80.59%   80.63%   +0.04%     
==========================================
  Files          62       62              
  Lines        5951     5965      +14     
==========================================
+ Hits         4796     4810      +14     
  Misses       1155     1155              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.