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]