def test_get_probability_within_distance_2(self):
distogram = Distogram('test')
distogram.original_file_format = 'pdb'
distance = Distance(36, 86, (1, ), ((6.589181, 6.589181), ), 0.934108)
distogram.add(distance)
self.assertEqual(1, distance.get_probability_within_distance(8))
self.assertEqual(0, distance.get_probability_within_distance(5))
def test_reshape_bins_2(self):
distogram = Distogram('test')
distogram.original_file_format = 'pdb'
distance = Distance(36, 86, (1, ), ((6.589181, 6.589181), ), 0.934108)
distogram.add(distance)
with self.assertRaises(ValueError):
distance.reshape_bins(((0, 1), (1, 10), (10, np.inf)))
def test_original_file_format_setter_2(self):
distogram = Distogram("test")
distogram.add(Distance(1, 25, (0.25, 0.45, 0.25, 0.05), ((0, 4), (4, 6), (6, 8), (8, np.inf))))
with self.assertRaises(ValueError):
distogram.original_file_format = 'mock_format'
def test_original_file_format_setter_1(self):
distogram = Distogram("test")
distogram.add(Distance(1, 25, (0.25, 0.45, 0.25, 0.05), ((0, 4), (4, 6), (6, 8), (8, np.inf))))
distogram.original_file_format = 'mmcif'
def test_predicted_distance_2(self):
distogram = Distogram('test')
distogram.original_file_format = 'pdb'
distance = Distance(36, 86, (1, ), ((6.589181, 6.589181), ), 0.934108)
distogram.add(distance)
self.assertEqual(6.589181, distance.predicted_distance)
