Add test for set_dice