From 02472c6f7c88c0cac507a81d930329f6ff688142 Mon Sep 17 00:00:00 2001
From: Johanna <johanna@zijderveld.de>
Date: Fri, 12 Apr 2024 12:31:18 +0200
Subject: [PATCH] undo incorrect formatting
---
codes/tb/test_tb.py | 21 ++++++++++++++++-----
1 file changed, 16 insertions(+), 5 deletions(-)
diff --git a/codes/tb/test_tb.py b/codes/tb/test_tb.py
index cb33995..a4d442b 100644
--- a/codes/tb/test_tb.py
+++ b/codes/tb/test_tb.py
@@ -1,7 +1,7 @@
# %%
import numpy as np
from codes.tb.tb import compareDicts
-from codes.kwant_helper import utils
+from codes.kwant_helper import utils
import itertools as it
from codes.tb.transforms import kfunc2tb, tb2kfunc, tb2kham, tb2khamvector
import pytest
@@ -24,10 +24,21 @@ def test_fourier():
tb_new = kfunc2tb(kfunc, nK, ndim=ndim)
compareDicts(h_0, tb_new)
-@pytest.mark.repeat(repeatNumber):
-def test_tbkham_transform():
- vectors = ((0, 0), (1, 0), (-1, 0), (0, 1), (0, -1), (1, -1), (-1, 1), (1, 1), (-1, -1))
+
+@pytest.mark.repeat(repeatNumber)
+def test_tbkham_transform():
+ vectors = (
+ (0, 0),
+ (1, 0),
+ (-1, 0),
+ (0, 1),
+ (0, -1),
+ (1, -1),
+ (-1, 1),
+ (1, 1),
+ (-1, -1),
+ )
ndof = 10
h_0 = utils.generate_guess(vectors, ndof)
- assert np.allclose(tb2kham(h_0, nK=nK, ndim=2), tb2khamvector(h_0, nK=nK, ndim=2))
\ No newline at end of file
+ assert np.allclose(tb2kham(h_0, nK=nK, ndim=2), tb2khamvector(h_0, nK=nK, ndim=2))
--
GitLab