Add tests for `select` function in `utils.py`

[Vinay Ramasesh]

The select function takes an array of shape (number_of_sequences, sequence_length, element_dimension) and selects a particular sequence element from each sequence.

We should write basic tests for this.