(get_clients)
| 111 | |
| 112 | |
| 113 | def test_stack(get_clients): |
| 114 | clients = get_clients(2) |
| 115 | |
| 116 | x_secret = torch.Tensor([0.0, 1, -2, 3, -4]) |
| 117 | y_secret = torch.Tensor([-4, 3, -2, 1, 0.0]) |
| 118 | secret_stacked = torch.stack([x_secret, y_secret]) |
| 119 | |
| 120 | session = Session(parties=clients) |
| 121 | SessionManager.setup_mpc(session) |
| 122 | x = MPCTensor(secret=x_secret, session=session) |
| 123 | y = MPCTensor(secret=y_secret, session=session) |
| 124 | stacked = stack([x, y]) |
| 125 | |
| 126 | assert (secret_stacked == stacked.reconstruct()).all() |
| 127 | |
| 128 | |
| 129 | def test_cat(get_clients): |
nothing calls this directly
no test coverage detected