Skip to content
Snippets Groups Projects
Commit e5daf0b3 authored by Joseph Weston's avatar Joseph Weston
Browse files

prefer 'tosympy=False' rather than constructing explicit Model

This is more idiomatic.
parent c0112080
No related branches found
No related tags found
No related merge requests found
...@@ -138,8 +138,8 @@ def test_graphene_to_kwant(): ...@@ -138,8 +138,8 @@ def test_graphene_to_kwant():
syst_from_family = model_to_builder(family, norbs, lat_vecs, atom_coords, coeffs=None) syst_from_family = model_to_builder(family, norbs, lat_vecs, atom_coords, coeffs=None)
# Generate using a single Model object # Generate using a single Model object
g = sympy.Symbol('g', real=True) g = sympy.Symbol('g', real=True)
ham = hamiltonian_from_family(family, coeffs=[g]) # tosympy=False to return a BlochModel
ham = Model(hamiltonian=ham, momenta=family[0].momenta) ham = hamiltonian_from_family(family, coeffs=[g], tosympy=False)
syst_from_model = model_to_builder(ham, norbs, lat_vecs, atom_coords) syst_from_model = model_to_builder(ham, norbs, lat_vecs, atom_coords)
# Make the graphene Hamiltonian using kwant only # Make the graphene Hamiltonian using kwant only
...@@ -186,8 +186,8 @@ def test_graphene_to_kwant(): ...@@ -186,8 +186,8 @@ def test_graphene_to_kwant():
family = family + onsites family = family + onsites
syst_from_family = model_to_builder(family, norbs, lat_vecs, atom_coords, coeffs=None) syst_from_family = model_to_builder(family, norbs, lat_vecs, atom_coords, coeffs=None)
gs = list(sympy.symbols('g0:%d'%3, real=True)) gs = list(sympy.symbols('g0:%d'%3, real=True))
ham = hamiltonian_from_family(family, coeffs=gs) # tosympy=False to return a BlochModel
ham = Model(hamiltonian=ham, momenta=family[0].momenta) ham = hamiltonian_from_family(family, coeffs=gs, tosympy=False)
syst_from_model = model_to_builder(ham, norbs, lat_vecs, atom_coords) syst_from_model = model_to_builder(ham, norbs, lat_vecs, atom_coords)
def onsite_A(site, c1): def onsite_A(site, c1):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment