def test_iter1(): r = [1, 2] a = Atom(5, r) for i, o in enumerate(a): assert o.R == r[i] for i, o in enumerate(a.iter()): assert o.R == r[i]
def test_iter2(): r = [1, 1, 2, 2] a = Atom(5, r) for i, o in enumerate(a.iter(True)): assert len(o) == 2