From ea48261fb63939b8eda4c4b0604f5037fba05a21 Mon Sep 17 00:00:00 2001
From: Christoph Groth <christoph.groth@cea.fr>
Date: Mon, 10 Apr 2017 11:26:58 +0200
Subject: [PATCH] discretizer: stop printing separator lines after the last
item
---
kwant/continuum/discretizer.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/kwant/continuum/discretizer.py b/kwant/continuum/discretizer.py
index f7a7ef1d..5e3eab13 100644
--- a/kwant/continuum/discretizer.py
+++ b/kwant/continuum/discretizer.py
@@ -216,8 +216,13 @@ def build_discretized(tb_hamiltonian, discrete_coordinates, *,
discrete_coordinates = sorted(discrete_coordinates)
tb = {}
+ first = True
for n, (offset, hopping) in enumerate(tb_hamiltonian.items()):
if verbose:
+ if first:
+ first = False
+ else:
+ print('\n')
print("Function generated for {}:".format(offset))
onsite = all(i == 0 for i in offset)
@@ -539,7 +544,7 @@ def _value_function(expr, discrete_coordinates, grid_spacing, onsite,
output = complex(expr)
if verbose:
- print("\n{}\n\n".format(output))
+ print("\n{}".format(output))
return output
@@ -565,7 +570,7 @@ def _value_function(expr, discrete_coordinates, grid_spacing, onsite,
if verbose:
for k, v in _cache.items():
print("\n{} = (\n{})".format(k, repr(np.array(v))))
- print('\n' + func_code + '\n\n')
+ print('\n' + func_code)
exec(func_code, namespace)
f = namespace[name]
--
GitLab