Skip to content
Snippets Groups Projects

Builder fixes

Merged Kostas Vilkelis requested to merge builder_fixes into main
1 file
+ 3
8
Compare changes
  • Side-by-side
  • Inline
@@ -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
Loading