Added unit tests for new die logic (that was tested elsewhere)