From 14e2cf6c181658a283abb3ae83104717afb9403d Mon Sep 17 00:00:00 2001 From: Pablo Piskunow <pablo.perez.piskunow@gmail.com> Date: Thu, 25 Jun 2020 15:30:11 +0200 Subject: [PATCH] test for special and general cases --- kwant/tests/test_builder.py | 26 ++++++++++++++------------ 1 file changed, 14 insertions(+), 12 deletions(-) diff --git a/kwant/tests/test_builder.py b/kwant/tests/test_builder.py index fc9a052f..46b3faee 100644 --- a/kwant/tests/test_builder.py +++ b/kwant/tests/test_builder.py @@ -863,18 +863,20 @@ def test_dangling(): sorted(site.tag for site in syst1.sites())) def test_dangling_with_symmetry(): - length = 3 - symm = kwant.TranslationalSymmetry((length, 0)) - lat = kwant.lattice.square(norbs=1) - syst = kwant.Builder(symmetry=symm) - for x in range(length): - syst[lat(x, 0)] = 0 - syst[lat(x, 1)] = 0 - syst[lat.neighbors()] = -1 - # remove neighbors of site at (lenght-1, 0), dangling across symm - del syst[lat(length - 2, 0)] - del syst[lat(length - 1, 1)] - syst.eradicate_dangling() + # lenght = 3 is the special case that gives all dangling bonds + # lenght = 4 is the standard case of dangling bonds across the symmetry + for length in [3, 4]: + symm = kwant.TranslationalSymmetry((length, 0)) + lat = kwant.lattice.square(norbs=1) + syst = kwant.Builder(symmetry=symm) + for x in range(length): + syst[lat(x, 0)] = 0 + syst[lat(x, 1)] = 0 + syst[lat.neighbors()] = -1 + # remove neighbors of site at (lenght-1, 0), dangling across symm + del syst[lat(length - 2, 0)] + del syst[lat(length - 1, 1)] + syst.eradicate_dangling() def test_builder_with_symmetry(): g = kwant.lattice.general(ta.identity(3), norbs=1) -- GitLab