Add test for die function.