Merge pull request #9 from Johnny-Wish/fix-test-logging

Fix test logging

Author: shuheng-liu

tests/test_function_basis.py

@@ -26,7 +26,7 @@ def test_legendre_polynomials():
         p2 = LegendrePolynomial(d)(x2)
         assert p2.requires_grad, f"output seems detached from the graph"
         p2 = p2.detach().cpu().numpy()
-        assert isclose(p2, p1).all(), f"p1 = {p1}, p2 = {p2}, delta = {p1 - p2}, max_delta = {abs(np.max(p1 - p2))}"
+        assert isclose(p2, p1).all(), f"p1 = {p1}, p2 = {p2}, delta = {p1 - p2}, max_delta = {np.max(abs(p1 - p2))}"
 
 
 def test_legendre_basis():
@@ -42,7 +42,7 @@ def test_legendre_basis():
     assert y2.requires_grad, f"output seems detached from the graph"
 
     y2 = y2.detach().cpu().numpy()
-    assert isclose(y2, y1).all(), f"y1 = {y1}, y2 = {y2}, delta = {y1 - y2}, max_delta = {abs(np.max(y1 - y2))}"
+    assert isclose(y2, y1).all(), f"y1 = {y1}, y2 = {y2}, delta = {y1 - y2}, max_delta = {np.max(abs(y1 - y2))}"
 
 
 def test_zero_order_spherical_harmonics():
@@ -66,7 +66,7 @@ def test_zero_order_spherical_harmonics():
     assert y2.requires_grad, f"output seems detached from the graph"
 
     y2 = y2.detach().cpu().numpy()
-    assert isclose(y2, y1).all(), f"y1 = {y1}, y2 = {y2}, delta = {y1 - y2}, max_delta = {abs(np.max(y1 - y2))}"
+    assert isclose(y2, y1).all(), f"y1 = {y1}, y2 = {y2}, delta = {y1 - y2}, max_delta = {np.max(abs(y1 - y2))}"
 
 
 def test_zero_order_spherical_harmonics_laplacian():
@@ -111,4 +111,4 @@ def laplacian1(u, r, theta, phi):
     lap1 = lap1.detach().cpu().numpy()
     lap2 = lap2.detach().cpu().numpy()
     assert isclose(lap2, lap1).all(), \
-        f"lap1 = {lap1}\nlap2 = {lap2}\ndelta = {lap1 - lap2}\nmax_delta = {abs(np.max(lap1 - lap2))}"
+        f"lap1 = {lap1}\nlap2 = {lap2}\ndelta = {lap1 - lap2}\nmax_delta = {np.max(abs(lap1 - lap2))}"