folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Solving #179 Simplex Domain Bug (#180)

Browse Source 
* simplex bug solved
---------

Co-authored-by: Dario Coscia <dariocoscia@dhcp-235.eduroam.sissa.it>
This commit is contained in:
Dario Coscia
2023-10-05 11:26:17 +02:00
committed by Nicola Demo
parent a9b1bd2826
commit 4850b0045d
2 changed files with 38 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
tests/test_geometry/test_simplex.py
View File

@@ -37,6 +37,33 @@ def test_constructor():
]
)
def test_sample():
# sampling inside
simplex = SimplexDomain(
[
LabelTensor(torch.tensor([[0, 0]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[1, 1]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[0, 2]]), labels=["x", "y"]),
]
)
pts = simplex.sample(10)
assert isinstance(pts, LabelTensor)
assert pts.size() == torch.Size([10, 2])
# sampling border
SimplexDomain(
[
LabelTensor(torch.tensor([[0, 0]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[1, 1]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[0, 2]]), labels=["x", "y"]),
],
sample_surface=True,
)
pts = simplex.sample(10)
assert isinstance(pts, LabelTensor)
assert pts.size() == torch.Size([10, 2])
def test_is_inside_faulty_point():
domain = SimplexDomain(