Adding ValueErrors and corresponding tests

Now raises ValueError when input is not in the right format. Also added tests to check if an error is actually raised when input is not in the right format.

Author: JasonMendoza2008

pelc/batch_eplet_comp.py

@@ -50,6 +50,10 @@ def compute_epletic_load(
     :return: None (if output_type is not None, the result will be saved on disk as a csv), or pandas.DataFrame
              (OutputType.COUNT_AND_DETAILS) or pandas.Series (OutputType.COUNT, or OutputType.ONLY_DETAILS) or
              tuple[pandas.DataFrame, pandas.DataFrame] (OutputType.FILTERED_TYPINGS)
+
+    :raises ValueError: if the number of unknown alleles is different for one donor and recipient pair or one allele is
+                        unknown whilst the other of the same locus isn't. Obviously doesn't raise anything if user
+                        inputs a homozygous like this: donor: A*01:01 A* and recipient: A*01:01 A*11:01.
     """
     if not class_i and not class_ii:
         logging.error(
@@ -74,7 +78,10 @@ def compute_epletic_load(
                 "Either the number of unknown alleles is different for one donor and recipient pair or one allele is "
                 "unknown whilst the other of the same locus isn't."
             )
-            return None
+            raise ValueError(
+                "Either the number of unknown alleles is different for one donor and recipient pair or one allele is "
+                "unknown whilst the other of the same locus isn't."
+            )
 
 
     df_a: pd.DataFrame

tests/pytest_only_one_chromosome_not_ok.xlsx

@@ -0,0 +1,68 @@
+import os
+import pandas as pd
+import pytest
+
+from pelc.batch_eplet_comp import compute_epletic_load
+from pelc.output_type import OutputType
+from tests.base_loading_for_tests import base_loading
+
+def test_one_chromosome() -> None:
+    """
+    Test the epletic load calculation with only one chromosome. If only one chromosome is present, it has to be
+    the first one, and it will mean it's a homozygote. Otherwise, _equal_amount_of_unknown_alleles will return False
+    and compute_epletic_load will return None. No file will then be created.
+    :return: None
+    """
+
+    # OK
+    donordf, recipientdf, output_path = base_loading("pytest_only_one_chromosome_ok.xlsx", "Sheet 1")
+
+    for class_1 in False, True:
+        compute_epletic_load(
+            donordf,
+            recipientdf,
+            output_path,
+            OutputType.DETAILS_AND_COUNT,
+            class_1,  # class_i
+            True,  # class_ii
+            False,  # abv_only
+        )
+
+        output_df: pd.DataFrame = pd.read_csv(f"{output_path}.csv", index_col="Index")
+
+        assert output_df.at[8, "EpMismatches"] == "160S_DQ"
+        assert output_df.at[8, "Eplet Load"] == 1
+
+        os.remove(f"{output_path}.csv")
+
+    # NOT OK
+    donordf, recipientdf, output_path = base_loading("pytest_only_one_chromosome_not_ok.xlsx", "Sheet 1")
+
+    for class_1 in False, True:
+        # Make sure this raises a ValueError
+        with pytest.raises(ValueError):
+            compute_epletic_load(
+                donordf,
+                recipientdf,
+                output_path,
+                OutputType.DETAILS_AND_COUNT,
+                class_1,  # class_i
+                True,  # class_ii
+                False,  # abv_only
+            )
+
+    # NOT OK2
+    donordf, recipientdf, output_path = base_loading("pytest_only_one_chromosome_not_ok_2.xlsx", "Sheet 1")
+
+    for class_1 in False, True:
+        # Make sure this raises a ValueError
+        with pytest.raises(ValueError):
+            compute_epletic_load(
+                donordf,
+                recipientdf,
+                output_path,
+                OutputType.DETAILS_AND_COUNT,
+                class_1,  # class_i
+                True,  # class_ii
+                False,  # abv_only
+            )