def test_lattice_iteration():
lattice = Lattice((48, 4), 2)
assert len(lattice) == (48 * 4 * 2)
for i, s in enumerate(lattice):
assert lattice[i] == s
assert lattice.index(s) == i
assert s in lattice
assert i + 1 == len(lattice)
with pytest.raises(ValueError):
lattice[len(lattice)]
assert LatticeSite((48, 5)) not in lattice
assert LatticeSite((46, 2), 2) not in lattice
assert lattice.count(LatticeSite((5, 3))) == 1
assert lattice.count(LatticeSite((245, 2))) == 0
