Adding primary selection unit tests