ARS_as_lg as a config flag to allow **ARS** suffix instead of **g** for lg reduction mode.

Author: pbashyal-nmdp

pyard/ard.py

@@ -54,6 +54,7 @@
     "ping": False,
     "map_drb345_to_drbx": True,
     "verbose_log": True,
+    "ARS_as_lg": False,
 }
 
 
@@ -64,7 +65,7 @@ class ARD(object):
     """
     ARD reduction for HLA
     Allows reducing alleles, allele code(MAC), Serology to
-    G, lg, lgx, W, exon and U2 levels.
+    G, lg, lgx, W, exon, S and U2 levels.
     """
 
     def __init__(
@@ -214,6 +215,8 @@ def _redux_allele(
                 # return allele with only first 2 field
                 redux_allele = ":".join(allele.split(":")[0:2])
             if redux_type == "lg":
+                if self._config["ARS_as_lg"]:
+                    return redux_allele + "ARS"
                 # lg mode has g appended with lgx reduction
                 return redux_allele + "g"
             return redux_allele

tests/environment.py

@@ -26,7 +26,13 @@ def before_all(context):
     context.ard = pyard.init("3440", data_dir="/tmp/py-ard")
 
     # an ard with ping set to True
-    my_config = {
+    ping_config = {
         "ping": True,
     }
-    context.ard_ping = pyard.init("3440", data_dir="/tmp/py-ard", config=my_config)
+    context.ard_ping = pyard.init("3440", data_dir="/tmp/py-ard", config=ping_config)
+
+    # an ard with ARS suffix for lg
+    lg_ars_config = {
+        "ARS_as_lg": True,
+    }
+    context.ard_ars = pyard.init("3440", data_dir="/tmp/py-ard", config=lg_ars_config)

tests/features/allele.feature

@@ -40,3 +40,20 @@ Feature: Alleles
       | C*02:10        | lg    | C*02:02g          |
       | C*02:10        | lgx   | C*02:02           |
       | C*06:17        | lgx   | C*06:17           |
+
+
+  Scenario Outline: allele reduction with ARS suffix
+
+    In `g` mode, use `ARS` prefix rather than `g`.
+
+    Given the allele as <Allele>
+    When reducing on the <Level> level with ARS suffix enabled
+    Then the reduced allele is found to be <Redux Allele>
+
+    Examples:
+      | Allele         | Level | Redux Allele   |
+      | A*01:01:01     | lg     | A*01:01ARS     |
+      | HLA-A*01:01:01 | lg     | HLA-A*01:01ARS |
+      | DRB1*14:06:01  | lg     | DRB1*14:06ARS  |
+      | C*02:02        | lg     | C*02:02ARS     |
+      | C*02:10        | lg     | C*02:02ARS     |

tests/steps/redux_allele.py

@@ -42,6 +42,12 @@ def step_impl(context, level):
     context.redux_allele = context.ard_ping.redux(context.allele, level)
 
 
+@when("reducing on the {level} level with ARS suffix enabled")
+def step_impl(context, level):
+    context.level = level
+    context.redux_allele = context.ard_ars.redux(context.allele, level)
+
+
 @when("reducing on the {level} level (ambiguous)")
 def step_impl(context, level):
     context.level = level