def test_info_multiple_edges(): info = get_info_from_edges(['O_K', 'N_K', 'Cr_L3']) assert len(info) == 3
def test_info_wrong_edge_format(): with pytest.raises(ValueError): get_info_from_edges(['O_K', 'NK'])
def test_info_one_edge(): info = get_info_from_edges('O_K') assert len(info) == 1