Added a function `test_shift()`

`test_shift()` improves the original `get_pi_shift()` to
recognise args of form `arg = x + n*pi/12` as opposed to
`arg = n*pi/12` which is currently recognised.

`get_pi_shift()` will be updated to this after adding some
rigorous tests.