def test_build_alphabet_map_w_alphabet_id(self): """build_alphabet_map: returns correct dict when given alphabet_id """ expected = dict([ ("G", "G"), ("A", "G"), ("V", "G"), ("L", "G"), ("I", "G"), ("S", "G"), ("P", "G"), ("T", "G"), ("C", "G"), ("N", "G"), ("D", "G"), ("X", "G"), ("B", "G"), ("M", "M"), ("F", "M"), ("Y", "M"), ("W", "M"), ("Q", "M"), ("K", "M"), ("H", "M"), ("R", "M"), ("E", "M"), ("Z", "M"), ]) self.assertEqual(build_alphabet_map("size_2"), expected) self.assertEqual(build_alphabet_map("charge_3")["E"], "D") self.assertEqual(build_alphabet_map("charge_3")["B"], "A") self.assertEqual(build_alphabet_map("charge_3")["K"], "K")
def test_build_alphabet_map_handles_all_ids_and_defs_wo_error(self): """build_alphabet_map: handles all pre-defined alphabets w/o error""" for alphabet_id, alphabet_def in list(alphabets.items()): try: build_alphabet_map(alphabet_id=alphabet_id) except ValueError: raise AssertionError("Failed on id: %s" % alphabet_id) try: build_alphabet_map(alphabet_def=alphabet_def) except ValueError: raise AssertionError("Failed on def: %s" % str(alphabet_def))
def test_build_alphabet_map_w_alphabet_def(self): """build_alphabet_map: returns correct dict when given alphabet_def """ expected = dict([ ("G", "S"), ("A", "S"), ("V", "S"), ("L", "S"), ("I", "S"), ("S", "S"), ("P", "S"), ("T", "S"), ("C", "S"), ("N", "S"), ("D", "S"), ("X", "S"), ("B", "S"), ("M", "L"), ("F", "L"), ("Y", "L"), ("W", "L"), ("Q", "L"), ("K", "L"), ("H", "L"), ("R", "L"), ("E", "L"), ("Z", "L"), ]) self.assertEqual( build_alphabet_map( alphabet_def=[("S", "GAVLISPTCNDXB"), ("L", "MFYWQKHREZ")]), expected, )