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