From ae1aed024657a77522eff61ae6edfe3eb7ca58ad Mon Sep 17 00:00:00 2001
From: Joseph Weston <joseph@weston.cloud>
Date: Tue, 10 Sep 2019 14:33:29 +0200
Subject: [PATCH] fix value function in test to return a scalar
Previously the value function would return a single-element vector
(which is illegal); now it returns a scalar.
---
kwant/tests/test_builder.py | 2 +-
kwant/tests/test_operator.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/kwant/tests/test_builder.py b/kwant/tests/test_builder.py
index 7ba00b40..e0ed4bfe 100644
--- a/kwant/tests/test_builder.py
+++ b/kwant/tests/test_builder.py
@@ -1275,7 +1275,7 @@ def test_argument_passing():
# Some common, some different args for value functions
def onsite2(site, a, b):
- return site.pos + a + b
+ return site.pos[0] + a + b
def hopping2(site1, site2, a, c, b):
return a + b + c
diff --git a/kwant/tests/test_operator.py b/kwant/tests/test_operator.py
index bb8a185e..c5111b75 100644
--- a/kwant/tests/test_operator.py
+++ b/kwant/tests/test_operator.py
@@ -445,7 +445,7 @@ def test_arg_passing(A):
lat1 = kwant.lattice.chain(norbs=1)
syst = kwant.Builder()
- syst[lat1(0)] = syst[lat1(1)] = lambda s0, a, b: s0.pos + a + b
+ syst[lat1(0)] = syst[lat1(1)] = lambda s0, a, b: s0.pos[0] + a + b
syst[lat1.neighbors()] = lambda s0, s1, a, b: a - b
fsyst = syst.finalized()
--
GitLab