Skip to content

Commit

Permalink
Small fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
mfherbst committed Nov 12, 2020
1 parent d836a88 commit 257eec6
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion adcc/AdcMatrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -368,7 +368,7 @@ def to_cpp(self):
def compute_apply(self, block, in_vec, out_vec):
self.innermatrix.compute_apply(block, in_vec, out_vec)
if block[0] == block[1]: # Diagonal block
out_vec = out_vec + self.shift * in_vec
out_vec += self.shift * in_vec

def diagonal(self, block):
out = self.innermatrix.diagonal(block)
Expand Down
8 changes: 4 additions & 4 deletions adcc/test_workflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -203,7 +203,7 @@ def test_diagonalise_adcmatrix(self):
matrix = adcc.AdcMatrix("adc2", adcc.LazyMp(cache.refstate["h2o_sto3g"]))

res = diagonalise_adcmatrix(matrix, n_states=3, kind="singlet",
solver_method="davidson")
eigensolver="davidson")
ref_singlets = refdata["adc2"]["singlet"]["eigenvalues"]
assert res.converged
assert res.eigenvalues == approx(ref_singlets[:3])
Expand All @@ -217,16 +217,16 @@ def test_diagonalise_adcmatrix(self):

with pytest.raises(InputError): # Too low tolerance
res = diagonalise_adcmatrix(matrix, n_states=9, kind="singlet",
solver_method="davidson",
eigensolver="davidson",
conv_tol=1e-14)

with pytest.raises(InputError): # Wrong solver method
res = diagonalise_adcmatrix(matrix, n_states=9, kind="singlet",
solver_method="blubber")
eigensolver="blubber")

with pytest.raises(InputError): # Too few guesses
res = diagonalise_adcmatrix(matrix, n_states=9, kind="singlet",
solver_method="davidson",
eigensolver="davidson",
guesses=guesses)

def test_estimate_n_guesses(self):
Expand Down

0 comments on commit 257eec6

Please sign in to comment.