diff --git a/meanfi/tests/test_kwant.py b/meanfi/tests/test_kwant.py index ab0a348cfd0a66169212d8a800e5dd91b219a1a2..48cb11709b3a80f6b046ecb3ec9c76175f1f6656 100644 --- a/meanfi/tests/test_kwant.py +++ b/meanfi/tests/test_kwant.py @@ -51,7 +51,6 @@ def test_kwant_supercell(seed): """Test with Kwant supercell and callable onsite and hoppings.""" np.random.seed(seed) ndim = np.random.randint(1, 3) - cutoff = np.random.randint(1, 3) sites_in_cell = np.random.randint(1, 4) ndof_per_site = [np.random.randint(1, 4) for site in range(sites_in_cell)] n_cells = np.random.randint(1, 4) @@ -108,20 +107,16 @@ 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 + raise \ No newline at end of file