def test_calculate_bins_empty_bins(self):
'''
>>> a = [0,3,6,6.1,9]
>>> for i in [0,20,40,60,80,100]:
... print(np.percentile(np.array(a),i))
...
0.0
2.4
4.8
6.04
6.68
9.0
'''
star_body = [
'{}.mrc {}'.format(str(i), str(i)) for i in [0, 3, 6, 6.0, 9]
]
img_list = [('{}.mrc'.format(i), float(i)) for i in [0, 3, 6, 6.0, 9]]
exp_5bins = {
0: ['0.mrc 0'],
1: ['3.mrc 3'],
2: ['6.mrc 6', '6.0.mrc 6.0'],
3: [],
4: ['9.mrc 9']
}
result_5bins = sh.calculate_bins(img_list, star_body, 5)
self.assertEqual(exp_5bins, result_5bins)
def test_calculate_bins_4bins(self):
"""
>>> a = np.array([0,1,2,3,4,5,6,7,8,9,10])
>>> for i in [0,25,50,75,100]:
... print(np.percentile(a, i))
...
0.0
2.5
5.0
7.5
10.0
"""
star_body = ["{}.mrc {}".format(str(i), str(i)) for i in [0, 3, 6, 9]]
img_list = [("{}.mrc".format(i), float(i)) for i in [0, 3, 6, 9]]
exp_4bins = {0: ["0.mrc 0"], 1: ["3.mrc 3"], 2: ["6.mrc 6"], 3: ["9.mrc 9"]}
result_4bins = sh.calculate_bins(img_list, star_body, 4)
self.assertEqual(exp_4bins, result_4bins)
def test_calculate_bins_empty_bins(self):
"""
>>> a = [0,3,6,6.1,9]
>>> for i in [0,20,40,60,80,100]:
... print(np.percentile(np.array(a),i))
...
0.0
2.4
4.8
6.04
6.68
9.0
"""
star_body = ["{}.mrc {}".format(str(i), str(i)) for i in [0, 3, 6, 6.0, 9]]
img_list = [("{}.mrc".format(i), float(i)) for i in [0, 3, 6, 6.0, 9]]
exp_5bins = {0: ["0.mrc 0"], 1: ["3.mrc 3"], 2: ["6.mrc 6", "6.0.mrc 6.0"], 3: [], 4: ["9.mrc 9"]}
result_5bins = sh.calculate_bins(img_list, star_body, 5)
self.assertEqual(exp_5bins, result_5bins)
def test_calculate_bins_4bins(self):
'''
>>> a = np.array([0,1,2,3,4,5,6,7,8,9,10])
>>> for i in [0,25,50,75,100]:
... print(np.percentile(a, i))
...
0.0
2.5
5.0
7.5
10.0
'''
star_body = ['{}.mrc {}'.format(str(i), str(i)) for i in [0, 3, 6, 9]]
img_list = [('{}.mrc'.format(i), float(i)) for i in [0, 3, 6, 9]]
exp_4bins = {
0: ['0.mrc 0'],
1: ['3.mrc 3'],
2: ['6.mrc 6'],
3: ['9.mrc 9']
}
result_4bins = sh.calculate_bins(img_list, star_body, 4)
self.assertEqual(exp_4bins, result_4bins)
def test_calculate_too_many_bins(self):
img_list = ['a', 'b']
with self.assertRaises(AssertionError):
sh.calculate_bins(img_list, self.body, 20)
def test_calculate_too_many_bins(self):
img_list = ["a", "b"]
with self.assertRaises(AssertionError):
sh.calculate_bins(img_list, self.body, 20)
