File size: 406 Bytes
64c992d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | import os
import sys
HERE = os.path.dirname(os.path.abspath(__file__))
sys.path.insert(0, HERE)
import spreadsheet as S # noqa: E402
def test_site_atom_indices_single():
assert S.site_atom_indices("21") == [20]
def test_site_atom_indices_multiple():
assert S.site_atom_indices("17,19") == [16, 18]
def test_site_atom_indices_accepts_int_input():
assert S.site_atom_indices(21) == [20]
|