remove back compatibility files for version 0.2
This commit is contained in:
giovanni
committed by
Giovanni Canali
Giovanni Canali
parent
ef3542486c
commit
684d691b78
184
tests/test_domain/test_simplex.py
Normal file
184
tests/test_domain/test_simplex.py
Normal file
@@ -0,0 +1,184 @@
|
||||
import torch
|
||||
import pytest
|
||||
|
||||
from pina import LabelTensor
|
||||
from pina.domain 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.0, -2.0]]), labels=["x", "y"]),
|
||||
LabelTensor(torch.tensor([[-0.5, -0.5]]), labels=["x", "y"]),
|
||||
LabelTensor(torch.tensor([[-2.0, 0.0]]), labels=["x", "y"]),
|
||||
LabelTensor(torch.tensor([[-0.5, 0.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
Block a user