def __init__(self, args: dict, user: str = None, password: str = None,
encoder: serializer.SerializerABC = serializer.JsonPickleSerializer,
user_dispatch: bool = True, download_models: bool = True,
dispatch_threshold: int = 0, model_threshold: int = (1024 ** 2) * 50):
super().__init__(args, user, password, user_dispatch)
self.args['download_models'] = download_models
self.args['dispatch_threshold'] = dispatch_threshold
self.args['model_threshold'] = model_threshold
self.model_encoder = encoder() if encoder else serializer.JsonPickleSerializer()
self.encoder = serializer.JsonPickleSerializer()
def test_serializer(self):
#Ensure serializability
notify = {'type': ffl.Notification.participant_joined}
s = serializer.JsonSerializer()
serialized = s.serialize(notify)
deserialized = s.deserialize(serialized)
self.assertTrue(ffl.Notification(deserialized['type']) is ffl.Notification.participant_joined)
s = serializer.JsonPickleSerializer()
serialized = s.serialize(notify)
deserialized = s.deserialize(serialized)
self.assertTrue(ffl.Notification(deserialized['type']) is ffl.Notification.participant_joined)
s = serializer.Base64Serializer()
serialized = s.serialize(notify)
deserialized = s.deserialize(serialized)
self.assertTrue(ffl.Notification(deserialized['type']) is ffl.Notification.participant_joined)
def test_numpy_serializer(self):
a = np.array([2, 3, 4])
serializer1 = serializer.JsonSerializer()
#Standard serializer should fail for numpy
with self.assertRaises(TypeError):
serializer1.serialize(a)
serializer2 = serializer.JsonPickleSerializer()
b = serializer2.serialize(a)
c = serializer2.deserialize(b)
self.assertTrue(np.array_equal(a, c))
d = serializer2.serialize(b)
e = serializer2.deserialize(d)
f = serializer2.deserialize(e)
self.assertTrue(np.array_equal(a, f))