Geometry Operations Enhancement (#122)
* updating exclusion domain - update sample/ is_inside - create tests * difference fixes - random iteration list for sample * created Intersection * created a Difference domain * unittest * docstrings and minor fixes * Refacotring Geometries - added OperationInterface - redid test cases - edited Union, Intersect, Exclusion, and Difference to inherit from OperationInterface - simplified Union, Intersect, Exclusion, and Difference * rm lighting logs --------- Co-authored-by: Dario Coscia <dariocoscia@cli-10-110-16-239.WIFIeduroamSTUD.units.it>
This commit is contained in:
51
tests/test_geometry/test_difference.py
Normal file
51
tests/test_geometry/test_difference.py
Normal file
@@ -0,0 +1,51 @@
|
||||
import torch
|
||||
|
||||
from pina import LabelTensor
|
||||
from pina.geometry import Difference, EllipsoidDomain, CartesianDomain
|
||||
|
||||
|
||||
def test_constructor_two_CartesianDomains():
|
||||
Difference([CartesianDomain({'x': [0, 2], 'y': [0, 2]}),
|
||||
CartesianDomain({'x': [1, 3], 'y': [1, 3]})])
|
||||
|
||||
|
||||
def test_constructor_two_3DCartesianDomain():
|
||||
Difference([CartesianDomain({'x': [0, 2], 'y': [0, 2], 'z': [0, 2]}),
|
||||
CartesianDomain({'x': [1, 3], 'y': [1, 3], 'z': [1, 3]})])
|
||||
|
||||
|
||||
def test_constructor_three_CartesianDomains():
|
||||
Difference([CartesianDomain({'x': [0, 2], 'y': [0, 2]}), CartesianDomain(
|
||||
{'x': [1, 3], 'y': [1, 3]}), CartesianDomain({'x': [2, 4], 'y': [2, 4]})])
|
||||
|
||||
|
||||
def test_is_inside_two_CartesianDomains():
|
||||
pt_1 = LabelTensor(torch.tensor([[0.5, 0.5]]), ['x', 'y'])
|
||||
pt_2 = LabelTensor(torch.tensor([[-1, -0.5]]), ['x', 'y'])
|
||||
domain = Difference([CartesianDomain({'x': [0, 2], 'y': [0, 2]}),
|
||||
CartesianDomain({'x': [1, 3], 'y': [1, 3]})])
|
||||
assert domain.is_inside(pt_1) == True
|
||||
assert domain.is_inside(pt_2) == False
|
||||
|
||||
|
||||
def test_is_inside_two_3DCartesianDomain():
|
||||
pt_1 = LabelTensor(torch.tensor([[0.5, 0.5, 0.5]]), ['x', 'y', 'z'])
|
||||
pt_2 = LabelTensor(torch.tensor([[-1, -0.5, -0.5]]), ['x', 'y', 'z'])
|
||||
domain = Difference([CartesianDomain({'x': [0, 2], 'y': [0, 2], 'z': [
|
||||
0, 2]}), CartesianDomain({'x': [1, 3], 'y': [1, 3], 'z': [1, 3]})])
|
||||
assert domain.is_inside(pt_1) == True
|
||||
assert domain.is_inside(pt_2) == False
|
||||
|
||||
|
||||
def test_sample():
|
||||
n = 100
|
||||
domain = Difference([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
|
||||
|
||||
n = 105
|
||||
pts = domain.sample(n)
|
||||
assert isinstance(pts, LabelTensor)
|
||||
assert pts.shape[0] == n
|
||||
Reference in New Issue
Block a user