예제 #1
0
파일: test_atom.py 프로젝트: silsgs/sisl
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]
예제 #2
0
파일: test_atom.py 프로젝트: silsgs/sisl
def test_iter2():
    r = [1, 1, 2, 2]
    a = Atom(5, r)
    for i, o in enumerate(a.iter(True)):
        assert len(o) == 2