Add test for output dtype of entropy