Take and restore snapshot

[kubaplas]

Add take_snapshot() and restore_snapshot() methods to the test env. It will be an interesting addition to the possible testing methods.