From f72ffa2bb7d9a1c7a22d9ea28304c180b806bdc8 Mon Sep 17 00:00:00 2001
From: Michael Wimmer <wimmer@lorentz.leidenuniv.nl>
Date: Tue, 27 Aug 2013 17:13:07 +0200
Subject: [PATCH] make ldos raise NotImplementedError if only selfenergy is
present (like wave_function)
---
kwant/solvers/common.py | 4 ++--
kwant/solvers/tests/_test_sparse.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/kwant/solvers/common.py b/kwant/solvers/common.py
index 96b5f7de..a19aee3b 100644
--- a/kwant/solvers/common.py
+++ b/kwant/solvers/common.py
@@ -460,8 +460,8 @@ class SparseSolver(object):
for lead in fsys.leads:
if not hasattr(lead, 'modes'):
# TODO: fix this
- raise ValueError("ldos for leads with only self-energy "
- "is not implemented yet")
+ raise NotImplementedError("ldos for leads with only "
+ "self-energy is not implemented yet")
linsys, lead_info = \
self._make_linear_sys(fsys, xrange(len(fsys.leads)), energy, args)
diff --git a/kwant/solvers/tests/_test_sparse.py b/kwant/solvers/tests/_test_sparse.py
index 989f144c..1e62908e 100644
--- a/kwant/solvers/tests/_test_sparse.py
+++ b/kwant/solvers/tests/_test_sparse.py
@@ -393,7 +393,7 @@ def test_ldos(ldos):
np.array([1, 1]) / (2 * np.pi))
assert_raises(ValueError, ldos, fsys.precalculate(what='selfenergy'), 0)
fsys.leads[0] = LeadWithOnlySelfEnergy(fsys.leads[0])
- assert_raises(ValueError, ldos, fsys, 0)
+ assert_raises(NotImplementedError, ldos, fsys, 0)
def test_wavefunc_ldos_consistency(wave_function, ldos):
--
GitLab