Update SnapshortAgentTest.py