def test_double_band(self):
lattice = Lattice([8, 2])
bands = Bands([3, 1], [periodic, periodic])
assert Orbitals.from_description(bands, lattice) == MomentaOrbitals(lattice, [(0, 0), (1, 0), (7, 0), (0, 1)], [periodic, periodic])
def test_single_band(self):
lattice = Lattice([8])
bands = Bands([4], [antiperiodic])
assert Orbitals.from_description(bands, lattice) == MomentaOrbitals(lattice, [(0,), (1,), (7,), (6,)], [antiperiodic])