mirror of
https://github.com/Xevion/exercism.git
synced 2025-12-06 07:14:56 -06:00
77 lines
2.6 KiB
Python
77 lines
2.6 KiB
Python
import unittest
|
|
|
|
from protein_translation import proteins
|
|
|
|
|
|
# Tests adapted from `problem-specifications//canonical-data.json` @ v1.1.1
|
|
|
|
class ProteinTranslationTest(unittest.TestCase):
|
|
|
|
def test_AUG_translates_to_methionine(self):
|
|
self.assertEqual(proteins('AUG'), ['Methionine'])
|
|
|
|
def test_identifies_Phenylalanine_codons(self):
|
|
for codon in ['UUU', 'UUC']:
|
|
self.assertEqual(proteins(codon), ['Phenylalanine'])
|
|
|
|
def test_identifies_Leucine_codons(self):
|
|
for codon in ['UUA', 'UUG']:
|
|
self.assertEqual(proteins(codon), ['Leucine'])
|
|
|
|
def test_identifies_Serine_codons(self):
|
|
for codon in ['UCU', 'UCC', 'UCA', 'UCG']:
|
|
self.assertEqual(proteins(codon), ['Serine'])
|
|
|
|
def test_identifies_Tyrosine_codons(self):
|
|
for codon in ['UAU', 'UAC']:
|
|
self.assertEqual(proteins(codon), ['Tyrosine'])
|
|
|
|
def test_identifies_Cysteine_codons(self):
|
|
for codon in ['UGU', 'UGC']:
|
|
self.assertEqual(proteins(codon), ['Cysteine'])
|
|
|
|
def test_identifies_Tryptophan_codons(self):
|
|
self.assertEqual(proteins('UGG'), ['Tryptophan'])
|
|
|
|
def test_identifies_stop_codons(self):
|
|
for codon in ['UAA', 'UAG', 'UGA']:
|
|
self.assertEqual(proteins(codon), [])
|
|
|
|
def test_translates_rna_strand_into_correct_protein_list(self):
|
|
strand = 'AUGUUUUGG'
|
|
expected = ['Methionine', 'Phenylalanine', 'Tryptophan']
|
|
self.assertEqual(proteins(strand), expected)
|
|
|
|
def test_stops_translation_if_stop_codon_at_beginning_of_sequence(self):
|
|
strand = 'UAGUGG'
|
|
expected = []
|
|
self.assertEqual(proteins(strand), expected)
|
|
|
|
def test_stops_translation_if_stop_codon_at_end_of_two_codon_sequence(
|
|
self):
|
|
strand = 'UGGUAG'
|
|
expected = ['Tryptophan']
|
|
self.assertEqual(proteins(strand), expected)
|
|
|
|
def test_stops_translation_if_stop_codon_at_end_of_three_codon_sequence(
|
|
self):
|
|
strand = 'AUGUUUUAA'
|
|
expected = ['Methionine', 'Phenylalanine']
|
|
self.assertEqual(proteins(strand), expected)
|
|
|
|
def test_stops_translation_if_stop_codon_in_middle_of_three_codon_sequence(
|
|
self):
|
|
strand = 'UGGUAGUGG'
|
|
expected = ['Tryptophan']
|
|
self.assertEqual(proteins(strand), expected)
|
|
|
|
def test_stops_translation_if_stop_codon_in_middle_of_six_codon_sequence(
|
|
self):
|
|
strand = 'UGGUGUUAUUAAUGGUUU'
|
|
expected = ['Tryptophan', 'Cysteine', 'Tyrosine']
|
|
self.assertEqual(proteins(strand), expected)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|