Added tests for get_pi_shift