Revert "remove test pickle"

This reverts commit 5d283d11cd313ff9ceeaa3ea9765b0d33d4e8835.