First batch of unit tests