Added some unit tests for the puzzle container calculation method