ydeng/autoBIGS.engine
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed issue where multiple allele identifiers were not returned in output.

Browse Source
This commit is contained in:
Harrison Deng 2025-01-17 17:04:22 +00:00
parent e7c8c5bcf9
commit b89f24a3fa
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/automlst/engine/data/local/csv.py
View File

@ -11,8 +11,8 @@ def dict_loci_alleles_variants_from_loci(alleles_map: Mapping[str, Sequence[Alle
if len(alleles) == 1:
result_dict[loci] = alleles[0].allele_variant
else:
for allele in alleles:
result_locis = list()
for allele in alleles:
result_locis.append(allele.allele_variant)
result_dict[loci] = result_locis
return result_dict

9
tests/automlst/engine/data/local/test_csv.py
View File

@ -10,3 +10,12 @@ def test_dict_loci_alleles_variants_from_loci_single_loci_not_list():
for loci, variant in results.items():
assert isinstance(variant, str)
assert variant == "1"
def test_dict_loci_alleles_variants_from_loci_multi_loci_is_list():
alleles_map = {
"adk": [Allele("adk", "1", None), Allele("adk", "2", None)]
}
results = dict_loci_alleles_variants_from_loci(alleles_map)
for loci, variant in results.items():
assert isinstance(variant, list)
assert len(variant) == 2