`ChoiceRV` currently does not behave like its NumPy equivalent when its
`a` parameters is an `int` or an array with more than one dimension. We
implement the necessary changes so that it does.