def test_finds_frags_res(self):
guessed_res = nomenclature.guess_nomenclature_fragments(
self.BW_local_w_pdb,
self.BW_local_w_pdb.top,
fragments=self.fragments,
return_residue_idxs=True)
_np.testing.assert_array_equal(self.fragments[4], guessed_res)
def test_finds_frags_seq_as_str(self):
guessed_frags = nomenclature.guess_nomenclature_fragments(
self.BW_local_w_pdb.seq,
self.BW_local_w_pdb.top,
fragments=self.fragments,
)
_np.testing.assert_array_equal([4], guessed_frags)
def test_finds_frags(self):
guessed_frags = nomenclature.guess_nomenclature_fragments(
self.BW_local_w_pdb,
self.BW_local_w_pdb.top,
fragments=self.fragments,
verbose=True,
)
_np.testing.assert_array_equal([4], guessed_frags)
def test_finds_frags_nothing_None(self):
seq = "THISSENTENCEWILLNEVERALIGN"
guessed_frags = nomenclature.guess_nomenclature_fragments(
seq, self.BW_local_w_pdb.top, fragments=self.fragments, empty=None)
print(guessed_frags)
assert guessed_frags is None