Improved test assertion coverage