Add some tests for the puzzle generation code.