File: test_block_manager_selector_unit.py

package info (click to toggle)
python-parsl 2025.01.13%2Bds-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 12,072 kB
  • sloc: python: 23,817; makefile: 349; sh: 276; ansic: 45
file content (20 lines) | stat: -rw-r--r-- 737 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import pytest

from parsl.executors.high_throughput.manager_record import ManagerRecord
from parsl.executors.high_throughput.manager_selector import BlockIdManagerSelector


@pytest.mark.local
def test_sort_managers():
    ready_managers = {
        b'manager1': {'block_id': 1},
        b'manager2': {'block_id': None},
        b'manager3': {'block_id': 3},
        b'manager4': {'block_id': 2}
    }

    manager_list = {b'manager1', b'manager2', b'manager3', b'manager4'}
    expected_sorted_list = [b'manager2', b'manager1', b'manager4', b'manager3']
    manager_selector = BlockIdManagerSelector()
    sorted_managers = manager_selector.sort_managers(ready_managers, manager_list)
    assert sorted_managers == expected_sorted_list