Add unit tests for counting sort