Skip to content

Commit

Permalink
remove print
Browse files Browse the repository at this point in the history
  • Loading branch information
wxj6000 committed Jan 4, 2024
1 parent fd4d5f8 commit 7ae3481
Showing 1 changed file with 0 additions and 10 deletions.
10 changes: 0 additions & 10 deletions gpu4pyscf/dft/tests/test_ao_values.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,6 @@ def test_ao_sph_deriv0(self):
ao_cpu = cupy.asarray(ao)
ni = NumInt(xc='LDA')
ao_gpu = numint.eval_ao(ni, mol_sph, coords, deriv=0)
print('sph_deriv0', cupy.linalg.norm(ao_cpu - ao_gpu))
assert cupy.linalg.norm(ao_cpu - ao_gpu) < 1e-8

def test_ao_sph_deriv1(self):
Expand All @@ -66,7 +65,6 @@ def test_ao_sph_deriv1(self):
ao_cpu = cupy.asarray(ao)
ni = NumInt(xc='LDA')
ao_gpu = numint.eval_ao(ni, mol_sph, coords, deriv=1)
print('sph_deriv1', cupy.linalg.norm(ao_cpu - ao_gpu))
assert cupy.linalg.norm(ao_cpu - ao_gpu) < 1e-8

def test_ao_sph_deriv2(self):
Expand All @@ -75,7 +73,6 @@ def test_ao_sph_deriv2(self):
ao_cpu = cupy.asarray(ao)
ni = NumInt(xc='LDA')
ao_gpu = numint.eval_ao(ni, mol_sph, coords, deriv=2)
print('sph_deriv2', cupy.linalg.norm(ao_cpu - ao_gpu))
assert cupy.linalg.norm(ao_cpu - ao_gpu) < 1e-8

def test_ao_sph_deriv3(self):
Expand All @@ -84,7 +81,6 @@ def test_ao_sph_deriv3(self):
ao_cpu = cupy.asarray(ao)
ni = NumInt(xc='LDA')
ao_gpu = numint.eval_ao(ni, mol_sph, coords, deriv=3)
print('sph_deriv3', cupy.linalg.norm(ao_cpu - ao_gpu))
assert cupy.linalg.norm(ao_cpu - ao_gpu) < 1e-8

def test_ao_sph_deriv4(self):
Expand All @@ -93,7 +89,6 @@ def test_ao_sph_deriv4(self):
ao_cpu = cupy.asarray(ao)
ni = NumInt(xc='LDA')
ao_gpu = numint.eval_ao(ni, mol_sph, coords, deriv=4)
print('sph_deriv4', cupy.linalg.norm(ao_cpu - ao_gpu))
assert cupy.linalg.norm(ao_cpu - ao_gpu) < 1e-8

# cart mol
Expand All @@ -103,7 +98,6 @@ def test_ao_cart_deriv0(self):
ao_cpu = cupy.asarray(ao)
ni = NumInt(xc='LDA')
ao_gpu = numint.eval_ao(ni, mol_cart, coords, deriv=0)
print('cart_deriv0', cupy.linalg.norm(ao_cpu - ao_gpu))
assert cupy.linalg.norm(ao_cpu - ao_gpu) < 1e-8

def test_ao_cart_deriv1(self):
Expand All @@ -112,7 +106,6 @@ def test_ao_cart_deriv1(self):
ao_cpu = cupy.asarray(ao)
ni = NumInt(xc='LDA')
ao_gpu = numint.eval_ao(ni, mol_cart, coords, deriv=1)
print('cart_deriv1', cupy.linalg.norm(ao_cpu - ao_gpu))
assert cupy.linalg.norm(ao_cpu - ao_gpu) < 1e-8

def test_ao_cart_deriv2(self):
Expand All @@ -121,7 +114,6 @@ def test_ao_cart_deriv2(self):
ao_cpu = cupy.asarray(ao)
ni = NumInt(xc='LDA')
ao_gpu = numint.eval_ao(ni, mol_cart, coords, deriv=2)
print('cart_deriv2', cupy.linalg.norm(ao_cpu - ao_gpu))
assert cupy.linalg.norm(ao_cpu - ao_gpu) < 1e-8

def test_ao_cart_deriv3(self):
Expand All @@ -130,7 +122,6 @@ def test_ao_cart_deriv3(self):
ao_cpu = cupy.asarray(ao)
ni = NumInt(xc='LDA')
ao_gpu = ni.eval_ao(mol_cart, coords, deriv=3)
print('cart_deriv3', cupy.linalg.norm(ao_cpu - ao_gpu))
assert cupy.linalg.norm(ao_cpu - ao_gpu) < 1e-8

def test_ao_cart_deriv4(self):
Expand All @@ -139,7 +130,6 @@ def test_ao_cart_deriv4(self):
ao_cpu = cupy.asarray(ao)
ni = NumInt(xc='LDA')
ao_gpu = numint.eval_ao(ni, mol_cart, coords, deriv=4)
print('cart_deriv4', cupy.linalg.norm(ao_cpu - ao_gpu))
assert cupy.linalg.norm(ao_cpu - ao_gpu) < 1e-8

if __name__ == "__main__":
Expand Down

0 comments on commit 7ae3481

Please sign in to comment.