fix: change int weights to float weights, update doctest examples (#61)

Closes #59.
fulcrumgenomics · Oct 4, 2024 · f71dc65 · f71dc65
1 parent a132a52
commit f71dc65
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/prymer/primer3/primer3_input.py b/prymer/primer3/primer3_input.py
@@ -64,8 +64,8 @@
 PRIMER_MAX_POLY_X -> 5
 PRIMER_MAX_NS_ACCEPTED -> 1
 PRIMER_LOWERCASE_MASKING -> 1
-PRIMER_PAIR_WT_PRODUCT_SIZE_LT -> 1
-PRIMER_PAIR_WT_PRODUCT_SIZE_GT -> 1
+PRIMER_PAIR_WT_PRODUCT_SIZE_LT -> 1.0
+PRIMER_PAIR_WT_PRODUCT_SIZE_GT -> 1.0
 PRIMER_PAIR_WT_PRODUCT_TM_LT -> 0.0
 PRIMER_PAIR_WT_PRODUCT_TM_GT -> 0.0
 PRIMER_WT_END_STABILITY -> 0.25

diff --git a/prymer/primer3/primer3_weights.py b/prymer/primer3/primer3_weights.py
@@ -42,14 +42,14 @@ class Primer3Weights:
 
      Example:
          >>> Primer3Weights() #default implementation
-         Primer3Weights(product_size_lt=1, product_size_gt=1, product_tm_lt=0.0, product_tm_gt=0.0, primer_end_stability=0.25, primer_gc_lt=0.25, primer_gc_gt=0.25, primer_self_any=0.1, primer_self_end=0.1, primer_size_lt=0.5, primer_size_gt=0.1, primer_tm_lt=1.0, primer_tm_gt=1.0)
+         Primer3Weights(product_size_lt=1.0, product_size_gt=1.0, product_tm_lt=0.0, product_tm_gt=0.0, primer_end_stability=0.25, primer_gc_lt=0.25, primer_gc_gt=0.25, primer_self_any=0.1, primer_self_end=0.1, primer_size_lt=0.5, primer_size_gt=0.1, primer_tm_lt=1.0, primer_tm_gt=1.0)
 
-         >>> Primer3Weights(product_size_lt=5)
-         Primer3Weights(product_size_lt=5, product_size_gt=1, product_tm_lt=0.0, product_tm_gt=0.0, primer_end_stability=0.25, primer_gc_lt=0.25, primer_gc_gt=0.25, primer_self_any=0.1, primer_self_end=0.1, primer_size_lt=0.5, primer_size_gt=0.1, primer_tm_lt=1.0, primer_tm_gt=1.0)
+         >>> Primer3Weights(product_size_lt=5.0)
+         Primer3Weights(product_size_lt=5.0, product_size_gt=1.0, product_tm_lt=0.0, product_tm_gt=0.0, primer_end_stability=0.25, primer_gc_lt=0.25, primer_gc_gt=0.25, primer_self_any=0.1, primer_self_end=0.1, primer_size_lt=0.5, primer_size_gt=0.1, primer_tm_lt=1.0, primer_tm_gt=1.0)
     """  # noqa: E501
 
-    product_size_lt: int = 1
-    product_size_gt: int = 1
+    product_size_lt: float = 1.0
+    product_size_gt: float = 1.0
     product_tm_lt: float = 0.0
     product_tm_gt: float = 0.0
     primer_end_stability: float = 0.25