Add test assertion