Python ShareTensor.__str__ Examples

Example #1

File: share_tensor_test.py Project: libratiger/SyMPC

def test_share_repr() -> None:

    x = torch.Tensor([5.0])
    x_share = ShareTensor(data=x)

    encoded_x = x_share.fp_encoder.encode(x)

    expected = f"[ShareTensor]\n\t| {x_share.fp_encoder}"
    expected = f"{expected}\n\t| Data: {encoded_x}"

    assert expected == x_share.__str__() == x_share.__repr__()

Example #2

def test_share_print() -> None:

    x = torch.Tensor([5.0])
    x_share = ShareTensor(data=x)

    encoded_x = x_share.fp_encoder.encode(x)

    expected = "[ShareTensor]"
    expected = f"{expected}\n\t| Session UUID: None"
    expected = f"{expected}\n\t| {x_share.fp_encoder}"
    expected = f"{expected}\n\t| Data: {encoded_x}"

    assert expected == x_share.__str__()