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()
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()