Add conversions between Codon and CodonAmbiguous

SecureDNA · Oct 17, 2023 · b57239e · b57239e
1 parent cb2ee54
commit b57239e
Showing 1 changed file with 51 additions and 2 deletions.
diff --git a/src/nucleotide.rs b/src/nucleotide.rs
@@ -435,6 +435,23 @@ impl CodonAmbiguous {
     }
 }
 
+impl From<Codon> for CodonAmbiguous {
+    #[inline(always)]
+    fn from(codon: Codon) -> Self {
+        Self(codon.0.map(Into::into))
+    }
+}
+
+impl TryFrom<CodonAmbiguous> for Codon {
+    type Error = TranslationError;
+
+    #[inline(always)]
+    fn try_from(codon: CodonAmbiguous) -> Result<Self, Self::Error> {
+        let [nuc0, nuc1, nuc2] = codon.0.map(TryInto::try_into);
+        Ok(Self([nuc0?, nuc1?, nuc2?]))
+    }
+}
+
 #[cfg(feature = "serde")]
 serde_utils::impl_stringlike!(Codon);
 
@@ -443,11 +460,11 @@ serde_utils::impl_stringlike!(CodonAmbiguous);
 
 #[cfg(test)]
 mod tests {
+    use super::*;
+
     #[cfg(feature = "serde")]
     #[test]
     fn test_serde_json() {
-        use super::*;
-
         assert_eq!(
             serde_json::to_value(Nucleotide::A).unwrap(),
             serde_json::json!("A")
@@ -470,4 +487,36 @@ mod tests {
             serde_json::json!("ABC")
         );
     }
+
+    #[test]
+    fn concrete_codon_to_ambiguous_codon_conversion() {
+        let codon = Codon([Nucleotide::C, Nucleotide::A, Nucleotide::T]);
+        let expected = CodonAmbiguous([
+            NucleotideAmbiguous::C,
+            NucleotideAmbiguous::A,
+            NucleotideAmbiguous::T,
+        ]);
+        assert_eq!(CodonAmbiguous::from(codon), expected);
+    }
+
+    #[test]
+    fn ambiguous_codon_to_concrete_codon_conversion() {
+        let amb_codon = CodonAmbiguous([
+            NucleotideAmbiguous::C,
+            NucleotideAmbiguous::A,
+            NucleotideAmbiguous::T,
+        ]);
+        let expected = Codon([Nucleotide::C, Nucleotide::A, Nucleotide::T]);
+        assert_eq!(Codon::try_from(amb_codon).unwrap(), expected);
+    }
+
+    #[test]
+    fn ambiguous_codon_to_concrete_codon_conversion_failure() {
+        let amb_codon = CodonAmbiguous([
+            NucleotideAmbiguous::B,
+            NucleotideAmbiguous::A,
+            NucleotideAmbiguous::T,
+        ]);
+        assert!(Codon::try_from(amb_codon).is_err());
+    }
 }