From 2f078c269cc741208d8092ea4d802269db0df27a Mon Sep 17 00:00:00 2001
From: Joseph Weston <joseph@weston.cloud>
Date: Fri, 19 Jul 2019 11:29:57 +0200
Subject: [PATCH] provide clearer error message when a Builder is passed to
plot_2d_bands
Closes #196.
---
kwant/wraparound.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/kwant/wraparound.py b/kwant/wraparound.py
index 25217826..08efec8c 100644
--- a/kwant/wraparound.py
+++ b/kwant/wraparound.py
@@ -362,10 +362,14 @@ def plot_2d_bands(syst, k_x=31, k_y=31, params=None,
if not hasattr(syst, '_wrapped_symmetry'):
raise TypeError("Expecting a system that was produced by "
"'kwant.wraparound.wraparound'.")
- if not isinstance(syst, system.FiniteSystem):
+ if isinstance(syst, system.InfiniteSystem):
msg = ("All symmetry directions must be wrapped around: specify "
"'keep=None' when calling 'kwant.wraparound.wraparound'.")
raise TypeError(msg)
+ if isinstance(syst, builder.Builder):
+ msg = ("Expecting a finalized system: remember to finalize your "
+ "system with 'syst.finalized()'.")
+ raise TypeError(msg)
params = params or {}
lat_ndim, space_ndim = syst._wrapped_symmetry.periods.shape
--
GitLab