improved the test for snapshot.py