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

Fix Codacy Warnings (#477)

Browse Source 
---------

Co-authored-by: Dario Coscia <dariocos99@gmail.com>
This commit is contained in:
Filippo Olivo
2025-03-10 15:38:45 +01:00
committed by Nicola Demo
parent e3790e049a
commit 4177bfbb50
157 changed files with 3473 additions and 3839 deletions
Download Patch File Download Diff File Expand all files Collapse all files

137
tests/test_geometry/test_union.py
View File

@@ -5,111 +5,88 @@ from pina.domain import Union, EllipsoidDomain, CartesianDomain
def test_constructor_two_CartesianDomains():
Union([
CartesianDomain({
'x': [0, 1],
'y': [0, 1]
}),
CartesianDomain({
'x': [0.5, 2],
'y': [-1, 0.1]
})
])
Union(
[
CartesianDomain({"x": [0, 1], "y": [0, 1]}),
CartesianDomain({"x": [0.5, 2], "y": [-1, 0.1]}),
]
)
def test_constructor_two_EllipsoidDomains():
Union([
EllipsoidDomain({
'x': [-1, 1],
'y': [-1, 1],
'z': [-1, 1]
}),
EllipsoidDomain({
'x': [-0.5, 0.5],
'y': [-0.5, 0.5],
'z': [-0.5, 0.5]
})
])
Union(
[
EllipsoidDomain({"x": [-1, 1], "y": [-1, 1], "z": [-1, 1]}),
EllipsoidDomain(
{"x": [-0.5, 0.5], "y": [-0.5, 0.5], "z": [-0.5, 0.5]}
),
]
)
def test_constructor_EllipsoidDomain_CartesianDomain():
Union([
EllipsoidDomain({
'x': [-1, 1],
'y': [-1, 1]
}),
CartesianDomain({
'x': [-0.5, 0.5],
'y': [-0.5, 0.5]
})
])
Union(
[
EllipsoidDomain({"x": [-1, 1], "y": [-1, 1]}),
CartesianDomain({"x": [-0.5, 0.5], "y": [-0.5, 0.5]}),
]
)
def test_is_inside_two_CartesianDomains():
pt_1 = LabelTensor(torch.tensor([[0.5, 0.5]]), ['x', 'y'])
pt_2 = LabelTensor(torch.tensor([[-1, -1]]), ['x', 'y'])
domain = Union([
CartesianDomain({
'x': [0, 1],
'y': [0, 1]
}),
CartesianDomain({
'x': [0.5, 2],
'y': [-1, 0.1]
})
])
pt_1 = LabelTensor(torch.tensor([[0.5, 0.5]]), ["x", "y"])
pt_2 = LabelTensor(torch.tensor([[-1, -1]]), ["x", "y"])
domain = Union(
[
CartesianDomain({"x": [0, 1], "y": [0, 1]}),
CartesianDomain({"x": [0.5, 2], "y": [-1, 0.1]}),
]
)
assert domain.is_inside(pt_1) == True
assert domain.is_inside(pt_2) == False
def test_is_inside_two_EllipsoidDomains():
pt_1 = LabelTensor(torch.tensor([[0.5, 0.5, 0.5]]), ['x', 'y', 'z'])
pt_2 = LabelTensor(torch.tensor([[-1, -1, -1]]), ['x', 'y', 'z'])
domain = Union([
EllipsoidDomain({
'x': [-1, 1],
'y': [-1, 1],
'z': [-1, 1]
}),
EllipsoidDomain({
'x': [-0.5, 0.5],
'y': [-0.5, 0.5],
'z': [-0.5, 0.5]
})
])
pt_1 = LabelTensor(torch.tensor([[0.5, 0.5, 0.5]]), ["x", "y", "z"])
pt_2 = LabelTensor(torch.tensor([[-1, -1, -1]]), ["x", "y", "z"])
domain = Union(
[
EllipsoidDomain({"x": [-1, 1], "y": [-1, 1], "z": [-1, 1]}),
EllipsoidDomain(
{"x": [-0.5, 0.5], "y": [-0.5, 0.5], "z": [-0.5, 0.5]}
),
]
)
assert domain.is_inside(pt_1) == True
assert domain.is_inside(pt_2) == False
def test_is_inside_EllipsoidDomain_CartesianDomain():
pt_1 = LabelTensor(torch.tensor([[0.5, 0.5]]), ['x', 'y'])
pt_2 = LabelTensor(torch.tensor([[-1, -1]]), ['x', 'y'])
domain = Union([
EllipsoidDomain({
'x': [-1, 1],
'y': [-1, 1],
}),
CartesianDomain({
'x': [0.6, 1.5],
'y': [-2, 0]
})
])
pt_1 = LabelTensor(torch.tensor([[0.5, 0.5]]), ["x", "y"])
pt_2 = LabelTensor(torch.tensor([[-1, -1]]), ["x", "y"])
domain = Union(
[
EllipsoidDomain(
{
"x": [-1, 1],
"y": [-1, 1],
}
),
CartesianDomain({"x": [0.6, 1.5], "y": [-2, 0]}),
]
)
assert domain.is_inside(pt_1) == True
assert domain.is_inside(pt_2) == False
def test_sample():
n = 100
domain = Union([
EllipsoidDomain({
'x': [-1, 1],
'y': [-1, 1]
}),
CartesianDomain({
'x': [-0.5, 0.5],
'y': [-0.5, 0.5]
})
])
domain = Union(
[
EllipsoidDomain({"x": [-1, 1], "y": [-1, 1]}),
CartesianDomain({"x": [-0.5, 0.5], "y": [-0.5, 0.5]}),
]
)
pts = domain.sample(n)
assert isinstance(pts, LabelTensor)
assert pts.shape[0] == n