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

Correct codacy warnings

Browse Source
This commit is contained in:
FilippoOlivo
2024-10-22 14:26:39 +02:00
committed by Nicola Demo
parent c9304fb9bb
commit 1bc1b3a580
15 changed files with 252 additions and 210 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_operators.py
View File

@@ -39,6 +39,7 @@ def test_grad_scalar_output():
]
assert torch.allclose(grad_tensor_s, true_val)
def test_grad_vector_output():
grad_tensor_v = grad(tensor_v, inp)
true_val = torch.cat(
@@ -75,6 +76,7 @@ def test_grad_vector_output():
]
assert torch.allclose(grad_tensor_v, true_val)
def test_div_vector_output():
div_tensor_v = div(tensor_v, inp)
true_val = 2*torch.sum(inp, dim=1).reshape(-1,1)
@@ -88,6 +90,7 @@ def test_div_vector_output():
assert div_tensor_v.labels == [f'dadx+dbdy']
assert torch.allclose(div_tensor_v, true_val)
def test_laplacian_scalar_output():
laplace_tensor_s = laplacian(tensor_s, inp)
true_val = 6*torch.ones_like(laplace_tensor_s)
@@ -101,6 +104,7 @@ def test_laplacian_scalar_output():
assert laplace_tensor_s.labels == [f"dd{tensor_s.labels[0]}"]
assert torch.allclose(laplace_tensor_s, true_val)
def test_laplacian_vector_output():
laplace_tensor_v = laplacian(tensor_v, inp)
true_val = 2*torch.ones_like(tensor_v)