[unit] Added more assertions