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