Add visual regression tests

[harunurhan]

With that we can remove enzyme shallow snapshot tests which are hard to work with.