Use snapbox for snapshot tests

[DaniPopes]

Instead of hardcoding e.g. flatten output, we should use the snapbox test framework