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_reshape_bins_1(self): distance = Distance(1, 25, (0.15, 0.45, 0.25, 0.05, 0.1), ((0, 4), (4, 6), (6, 8), (8, 10), (10, np.inf))) new_bins = ((0, 2), (2, 8), (8, np.inf)) distance.reshape_bins(new_bins) self.assertEqual(distance.raw_score, 0.85) self.assertEqual(round(distance.get_probability_within_distance(8), 2), 0.85) self.assertTupleEqual(new_bins, distance.distance_bins) self.assertTupleEqual((0.075, 0.775, 0.15000000000000002), distance.distance_scores)