From 9fdc0c28a4c3660ae58277ee9ed537e96f9134f4 Mon Sep 17 00:00:00 2001
From: antoniolrm <am@antoniomanesco.org>
Date: Fri, 17 May 2024 13:03:45 +0200
Subject: [PATCH] loop over all cells
---
meanfi/tests/test_kwant.py | 11 +++++------
1 file changed, 5 insertions(+), 6 deletions(-)
diff --git a/meanfi/tests/test_kwant.py b/meanfi/tests/test_kwant.py
index 17a9a63..c8bb1eb 100644
--- a/meanfi/tests/test_kwant.py
+++ b/meanfi/tests/test_kwant.py
@@ -27,14 +27,13 @@ def test_kwant_conversion(seed):
norbs=ndof_per_site,
)
- dummy_tb = kwant.Builder(
- kwant.TranslationalSymmetry(*n_cells * lattice.prim_vecs)
- )
+ dummy_tb = kwant.Builder(kwant.TranslationalSymmetry(*n_cells * lattice.prim_vecs))
for site in range(sites_in_cell):
for sublattice in lattice.sublattices:
- dummy_tb[sublattice(site, *[0 for _ in range(ndim - 1)])] = (
- np.eye(sum(ndof_per_site)) * 2
- )
+ for n in range(n_cells):
+ dummy_tb[sublattice(site, *[n for _ in range(ndim - 1)])] = (
+ np.eye(sum(ndof_per_site)) * 2
+ )
# generate random and generate builder from it
random_tb = guess_tb(keyList, sum(ndof_per_site) * sites_in_cell * n_cells)
--
GitLab