Add snapshot tests

[0x61nas]

[webbrain-one]

Hi! I've opened a pull request to address this: #124. This adds the missing snapshot tests to ensure visual consistency moving forward.