From 7b56e481f3cb4c1853c3635f7a8e51bcde664039 Mon Sep 17 00:00:00 2001
From: antoniolrm <am@antoniomanesco.org>
Date: Fri, 17 May 2024 18:01:13 +0200
Subject: [PATCH] we don't need to return data here
---
meanfi/tests/test_kwant.py | 12 ++++++++----
1 file changed, 8 insertions(+), 4 deletions(-)
diff --git a/meanfi/tests/test_kwant.py b/meanfi/tests/test_kwant.py
index 48cb117..3785f00 100644
--- a/meanfi/tests/test_kwant.py
+++ b/meanfi/tests/test_kwant.py
@@ -42,7 +42,7 @@ def test_kwant_conversion(seed):
random_tb, list(dummy_tb.sites()), dummy_tb.symmetry.periods
)
# convert builder back to tb and compare
- random_builder_tb = builder_to_tb(random_builder, return_data=True)
+ random_builder_tb = builder_to_tb(random_builder)
compare_dicts(random_tb, random_builder_tb)
@@ -107,16 +107,20 @@ def test_kwant_supercell(seed):
site1, site2 = site_pair[0]
if site1 == site2:
assert np.isclose(
- random_builder[site1](site=site1, alpha=params["alpha"], beta=params["beta"]),
+ random_builder[site1](
+ site=site1, alpha=params["alpha"], beta=params["beta"]
+ ),
random_builder_test[site1],
).all()
else:
try:
assert np.isclose(
- random_builder[site1, site2](site1, site2, gamma=params["gamma"], delta=params["delta"]),
+ random_builder[site1, site2](
+ site1, site2, gamma=params["gamma"], delta=params["delta"]
+ ),
random_builder_test[site1, site2],
).all()
except KeyError:
continue
except:
- raise
\ No newline at end of file
+ raise
--
GitLab