From add7d62e719323caec84020676f0eb739403c32d Mon Sep 17 00:00:00 2001
From: antoniolrm <am@antoniomanesco.org>
Date: Fri, 17 May 2024 13:45:09 +0200
Subject: [PATCH] assign site dof properly
---
meanfi/tests/test_kwant.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/meanfi/tests/test_kwant.py b/meanfi/tests/test_kwant.py
index c8bb1eb..9b947fa 100644
--- a/meanfi/tests/test_kwant.py
+++ b/meanfi/tests/test_kwant.py
@@ -29,10 +29,10 @@ def test_kwant_conversion(seed):
dummy_tb = kwant.Builder(kwant.TranslationalSymmetry(*n_cells * lattice.prim_vecs))
for site in range(sites_in_cell):
- for sublattice in lattice.sublattices:
+ for i, sublattice in enumerate(lattice.sublattices):
for n in range(n_cells):
dummy_tb[sublattice(site, *[n for _ in range(ndim - 1)])] = (
- np.eye(sum(ndof_per_site)) * 2
+ np.eye(ndof_per_site[i]) * 2
)
# generate random and generate builder from it
--
GitLab