From 5b2bc6a43961b217394915c138d1d333506577eb Mon Sep 17 00:00:00 2001
From: Joseph Weston <joseph@weston.cloud>
Date: Fri, 27 Sep 2019 14:00:12 +0200
Subject: [PATCH] add test for kpm.conductivity when 'norbs' not provided

---
 kwant/tests/test_kpm.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/kwant/tests/test_kpm.py b/kwant/tests/test_kpm.py
index 7ae18733..524cc57d 100644
--- a/kwant/tests/test_kpm.py
+++ b/kwant/tests/test_kpm.py
@@ -80,6 +80,13 @@ def test_conductivity():
         cond = kwant.kpm.conductivity(syst, alpha=alpha, beta=beta,
                                         positions=None, **kpm_params)
 
+    # test when 'norbs' not provided
+    n = lat.norbs
+    lat.norbs = None
+    cond = kwant.kpm.conductivity(syst, alpha=alpha, beta=beta,
+                                  positions=None, **kpm_params)
+    lat.norbs = n
+
     # test system or hamiltonian, no positions, but velocity operators
     cond_xx = kwant.kpm.conductivity(syst, alpha='x', beta='x',
                                      positions=None, **kpm_params)
-- 
GitLab