Python get_info_from_edges примеры использования

Пример #1

def test_info_multiple_edges():
    info = get_info_from_edges(['O_K', 'N_K', 'Cr_L3'])
    assert len(info) == 3

Пример #2

def test_info_wrong_edge_format():
    with pytest.raises(ValueError):
        get_info_from_edges(['O_K', 'NK'])

Пример #3

def test_info_one_edge():
    info = get_info_from_edges('O_K')
    assert len(info) == 1