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

src/automlst/engine/data/local/csv.py

@@ -1,7 +1,6 @@
 import csv
-from io import TextIOWrapper
 from os import PathLike
-from typing import AsyncIterable, Iterable, Mapping, Sequence, Union
+from typing import AsyncIterable, Mapping, Sequence, Union
 
 from automlst.engine.data.structures.mlst import Allele, MLSTProfile
 
@@ -11,10 +10,11 @@ def dict_loci_alleles_variants_from_loci(alleles_map: Mapping[str, Sequence[Alle
     for loci, alleles in alleles_map.items():
         if len(alleles) == 1:
             result_dict[loci] = alleles[0].allele_variant
-        for allele in alleles:
+        else:
             result_locis = list()
-            result_locis.append(allele.allele_variant)
+            for allele in alleles:
-            result_dict[loci] = result_locis
+                result_locis.append(allele.allele_variant)
+                result_dict[loci] = result_locis
     return result_dict
 
 

tests/automlst/engine/data/local/test_csv.py

@@ -0,0 +1,21 @@
+from automlst.engine.data.local.csv import dict_loci_alleles_variants_from_loci
+from automlst.engine.data.structures.mlst import Allele
+
+
+def test_dict_loci_alleles_variants_from_loci_single_loci_not_list():
+    alleles_map = {
+        "adk": [Allele("adk", "1", None)]
+    }
+    results = dict_loci_alleles_variants_from_loci(alleles_map)
+    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