Files
pymol-open-source/testing/tests/api/test_selecting.py
2025-12-08 09:32:07 -05:00

18 lines
524 B
Python

from pymol import cmd
def test_select_operators():
cmd.reinitialize()
cmd.pseudoatom(pos=[1, 2, 3], b=5)
assert cmd.count_atoms("b > 4") == 1
assert cmd.count_atoms("b < 6") == 1
assert cmd.count_atoms("b = 5") == 1
assert cmd.count_atoms("b >= 5") == 1
assert cmd.count_atoms("b <= 5") == 1
assert cmd.count_atoms("b > 4 & x == 1") == 1
assert cmd.count_atoms("b > 5 & x == 1") == 0
assert cmd.count_atoms("b < 6 & y <= 3") == 1
assert cmd.count_atoms("b = 5 & z >= 2") == 1