folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
6cfb8fd28b607f0051320485a7508e28057ef467
PINA/tests/test_geometry/test_simplex.py
Dario Coscia 8b7b61b3bd Documentation for v0.1 version (#199) 
* Adding Equations, solving typos
* improve _code.rst
* the team rst and restuctore index.rst
* fixing errors

---------

Co-authored-by: Dario Coscia <dariocoscia@dhcp-015.eduroam.sissa.it>
2023-11-17 09:51:29 +01:00

172 lines
6.9 KiB
Python
Raw Blame History

import torch
import pytest
from pina import LabelTensor
from pina.geometry import SimplexDomain
def test_constructor():
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"]),
])
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,
)
with pytest.raises(ValueError):
# different labels
SimplexDomain([
LabelTensor(torch.tensor([[0, 0]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[1, 1]]), labels=["x", "z"]),
LabelTensor(torch.tensor([[0, 2]]), labels=["x", "a"]),
])
# not LabelTensor
SimplexDomain([
LabelTensor(torch.tensor([[0, 0]]), labels=["x", "y"]),
[1, 1],
LabelTensor(torch.tensor([[0, 2]]), labels=["x", "y"]),
])
# different number of vertices
SimplexDomain([
LabelTensor(torch.tensor([[0., -2.]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[-.5, -.5]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[-2., 0.]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[-.5, .5]]), labels=["x", "y"]),
])
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(
[
LabelTensor(torch.tensor([[0, 0]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[2, 2]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[4, 0]]), labels=["x", "y"]),
],
sample_surface=True,
)
pt = LabelTensor(torch.tensor([[2, 1]]), ["x", "z"])
with pytest.raises(ValueError):
assert domain.is_inside(point=pt, check_border=False) == True
def test_is_inside_2D_check_border_true():
domain = SimplexDomain(
[
LabelTensor(torch.tensor([[0, 0]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[2, 2]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[4, 0]]), labels=["x", "y"]),
],
sample_surface=True,
)
pt1 = LabelTensor(torch.tensor([[0, 0]]), ["x", "y"])
pt2 = LabelTensor(torch.tensor([[2, 2]]), ["x", "y"])
pt3 = LabelTensor(torch.tensor([[4, 0]]), ["x", "y"])
pt4 = LabelTensor(torch.tensor([[3, 1]]), ["x", "y"])
pt5 = LabelTensor(torch.tensor([[2, 1]]), ["x", "y"])
pt6 = LabelTensor(torch.tensor([[100, 100]]), ["x", "y"])
pts = [pt1, pt2, pt3, pt4, pt5, pt6]
for pt, exp_result in zip(pts, [True, True, True, True, False, False]):
assert domain.is_inside(point=pt, check_border=True) == exp_result
def test_is_inside_2D_check_border_false():
domain = SimplexDomain(
[
LabelTensor(torch.tensor([[0, 0]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[2, 2]]), labels=["x", "y"]),
LabelTensor(torch.tensor([[4, 0]]), labels=["x", "y"]),
],
sample_surface=False,
)
pt1 = LabelTensor(torch.tensor([[0, 0]]), ["x", "y"])
pt2 = LabelTensor(torch.tensor([[2, 2]]), ["x", "y"])
pt3 = LabelTensor(torch.tensor([[4, 0]]), ["x", "y"])
pt4 = LabelTensor(torch.tensor([[3, 1]]), ["x", "y"])
pt5 = LabelTensor(torch.tensor([[2, 1]]), ["x", "y"])
pt6 = LabelTensor(torch.tensor([[2.5, 1]]), ["x", "y"])
pt7 = LabelTensor(torch.tensor([[100, 100]]), ["x", "y"])
pts = [pt1, pt2, pt3, pt4, pt5, pt6, pt7]
for pt, exp_result in zip(pts,
[False, False, False, False, True, True, False]):
assert domain.is_inside(point=pt, check_border=False) == exp_result
def test_is_inside_3D_check_border_true():
domain = SimplexDomain(
[
LabelTensor(torch.tensor([[0, 0, 0]]), labels=["x", "y", "z"]),
LabelTensor(torch.tensor([[2, 2, 0]]), labels=["x", "y", "z"]),
LabelTensor(torch.tensor([[4, 0, 0]]), labels=["x", "y", "z"]),
LabelTensor(torch.tensor([[0, 0, 20]]), labels=["x", "y", "z"]),
],
sample_surface=True,
)
pt1 = LabelTensor(torch.tensor([[0, 0, 0]]), ["x", "y", "z"])
pt2 = LabelTensor(torch.tensor([[2, 2, 0]]), ["x", "y", "z"])
pt3 = LabelTensor(torch.tensor([[4, 0, 0]]), ["x", "y", "z"])
pt4 = LabelTensor(torch.tensor([[3, 1, 0]]), ["x", "y", "z"])
pt5 = LabelTensor(torch.tensor([[2, 1, 0]]), ["x", "y", "z"])
pt6 = LabelTensor(torch.tensor([[100, 100, 0]]), ["x", "y", "z"])
pt7 = LabelTensor(torch.tensor([[0, 0, 19]]), ["x", "y", "z"])
pt8 = LabelTensor(torch.tensor([[0, 0, 20]]), ["x", "y", "z"])
pt9 = LabelTensor(torch.tensor([[2, 1, 1]]), ["x", "y", "z"])
pts = [pt1, pt2, pt3, pt4, pt5, pt6, pt7, pt8, pt9]
for pt, exp_result in zip(
pts, [True, True, True, True, True, False, True, True, False]):
assert domain.is_inside(point=pt, check_border=True) == exp_result
def test_is_inside_3D_check_border_false():
domain = SimplexDomain(
[
LabelTensor(torch.tensor([[0, 0, 0]]), labels=["x", "y", "z"]),
LabelTensor(torch.tensor([[2, 2, 0]]), labels=["x", "y", "z"]),
LabelTensor(torch.tensor([[4, 0, 0]]), labels=["x", "y", "z"]),
LabelTensor(torch.tensor([[0, 0, 20]]), labels=["x", "y", "z"]),
],
sample_surface=False,
)
pt1 = LabelTensor(torch.tensor([[0, 0, 0]]), ["x", "y", "z"])
pt2 = LabelTensor(torch.tensor([[3, 1, 0]]), ["x", "y", "z"])
pt3 = LabelTensor(torch.tensor([[2, 1, 0]]), ["x", "y", "z"])
pt4 = LabelTensor(torch.tensor([[100, 100, 0]]), ["x", "y", "z"])
pt5 = LabelTensor(torch.tensor([[0, 0, 19]]), ["x", "y", "z"])
pt6 = LabelTensor(torch.tensor([[0, 0, 20]]), ["x", "y", "z"])
pt7 = LabelTensor(torch.tensor([[2, 1, 1]]), ["x", "y", "z"])
pts = [pt1, pt2, pt3, pt4, pt5, pt6, pt7]
for pt, exp_result in zip(pts,
[False, False, False, False, False, False, True]):
assert domain.is_inside(point=pt, check_border=False) == exp_result
Reference in New Issue View Git Blame Copy Permalink