def test_trimesh():
pos = torch.tensor([[0, 0, 0], [1, 0, 0], [0, 1, 0], [1, 1, 0]],
dtype=torch.float)
face = torch.tensor([[0, 1, 2], [1, 2, 3]]).t()
data = Data(pos=pos, face=face)
mesh = to_trimesh(data)
data = from_trimesh(mesh)
assert pos.tolist() == data.pos.tolist()
assert face.tolist() == data.face.tolist()
def test_trimesh():
pos = torch.tensor([[0, 0, 0], [1, 0, 0], [0, 1, 0], [1, 1, 0]],
dtype=torch.float)
face = torch.tensor([[0, 1, 2], [1, 2, 3]]).t()
data = Data(pos=pos, face=face)
mesh = to_trimesh(data)
data = from_trimesh(mesh)
perm = (data.pos * torch.tensor([[1.0, 2.0, 3.0]])).sum(dim=-1).argsort()
assert pos.tolist() == data.pos[perm].tolist()
assert face.tolist() == perm[data.face].tolist()