Add unit test for Kryo serialization of EdgePartition