Beispiel #1
0
def test_send_get(clients, precision, base) -> None:
    x = torch.Tensor([0.122, 1.342, 4.67])
    x_share = ShareTensor(data=x,
                          encoder_precision=precision,
                          encoder_base=base)
    x_ptr = x_share.send(clients[0])

    assert x_share == x_ptr.get()
Beispiel #2
0
def test_send_get(get_clients, precision, base) -> None:
    x = torch.Tensor([0.122, 1.342, 4.67])
    x_share = ShareTensor(data=x,
                          config=Config(encoder_precision=precision,
                                        encoder_base=base))
    client = get_clients(1)[0]
    x_ptr = x_share.send(client)

    assert x_share == x_ptr.get()