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

fix tests

Browse Source
This commit is contained in:
Nicola Demo
2025-01-23 09:52:23 +01:00
parent 9aed1a30b3
commit a899327de1
32 changed files with 2331 additions and 2428 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_utils.py
View File

@@ -6,7 +6,7 @@ from pina import LabelTensor
from pina.domain import EllipsoidDomain, CartesianDomain
from pina.utils import check_consistency
import pytest
from pina.domain import Location
from pina.domain import DomainInterface
def test_merge_tensors():
@@ -27,8 +27,8 @@ def test_check_consistency_correct():
example_input_pts = LabelTensor(torch.tensor([[0, 0, 0]]), ['x', 'y', 'z'])
check_consistency(example_input_pts, torch.Tensor)
check_consistency(CartesianDomain, Location, subclass=True)
check_consistency(ellipsoid1, Location)
check_consistency(CartesianDomain, DomainInterface, subclass=True)
check_consistency(ellipsoid1, DomainInterface)
def test_check_consistency_incorrect():
@@ -36,8 +36,8 @@ def test_check_consistency_incorrect():
example_input_pts = LabelTensor(torch.tensor([[0, 0, 0]]), ['x', 'y', 'z'])
with pytest.raises(ValueError):
check_consistency(example_input_pts, Location)
check_consistency(example_input_pts, DomainInterface)
with pytest.raises(ValueError):
check_consistency(torch.Tensor, Location, subclass=True)
check_consistency(torch.Tensor, DomainInterface, subclass=True)
with pytest.raises(ValueError):
check_consistency(ellipsoid1, torch.Tensor)