class AngularDistanceTest(unittest.TestCase):
def setUp(self):
star1 = Star(0,1,sph=(0,0))
star2 = Star(1,1,sph=(np.pi,0))
star3 = Star(2,1,sph=(np.pi/3,0))
star4 = Star(3,1,sph=(np.pi/2,0))
self.d1 = AngularDistance(star1,star2)
self.d2 = AngularDistance(star1,star3)
self.d3 = AngularDistance(star2,star3)
self.d4 = AngularDistance(star1,star4)
def test_equal_to_self(self):
self.assertEquals(self.d1, self.d1)
def test_not_equal_to_others(self):
self.assertNotEquals(self.d1, self.d2)
self.assertNotEquals(self.d1, self.d3)
self.assertNotEquals(self.d1, self.d4)
def test_check_triplet_true(self):
self.assertEquals(self.d1.check_triplet(self.d2,self.d3), True)
def test_check_triplet_false(self):
self.assertEquals(self.d1.check_triplet(self.d2,self.d4), False)
def setUp(self):
star1 = Star(0,1,sph=(0,0))
star2 = Star(1,1,sph=(np.pi,0))
star3 = Star(2,1,sph=(np.pi/3,0))
star4 = Star(3,1,sph=(np.pi/2,0))
self.d1 = AngularDistance(star1,star2)
self.d2 = AngularDistance(star1,star3)
self.d3 = AngularDistance(star2,star3)
self.d4 = AngularDistance(star1,star4)
def test_get_all_angular_distances_sorted(self):
star1 = Star(1,1,sph=(0,np.pi/4))
star2 = Star(2,1,sph=(0,np.pi/2))
star3 = Star(3,1,sph=(0,np.pi))
distance1 = AngularDistance(star1, star2)
distance2 = AngularDistance(star1, star3)
distance3 = AngularDistance(star2, star3)
scene = Scene([star1,star2,star3])
distances = scene.get_all_angular_distances()
self.assertEquals(distance1,distances.distances[0])
self.assertEquals(distance3,distances.distances[1])
self.assertEquals(distance2,distances.distances[2])
def test_get_all_angular_distances_max(self):
star1 = self.star1
star2 = self.star2
star3 = self.star3
distance1 = AngularDistance(star1, star2)
distance2 = AngularDistance(star1, star3)
distance3 = AngularDistance(star2, star3)
scene = Scene([star1,star2,star3])
max_distance = 2
distances = scene.get_all_angular_distances(max_distance)
self.assertIn(distance1, distances.distances)
self.assertIn(distance2, distances.distances)
def setUp(self):
distances = Distances()
self.star1 = Star(1, 0, sph=(0, 0))
self.star2 = Star(2, 0, sph=(np.pi / 3, 0))
self.star3 = Star(3, 0, sph=(np.pi, 0))
self.star4 = Star(4, 0, sph=(2 * np.pi / 3 + 0.001, 0))
self.distance12 = AngularDistance(self.star1, self.star2)
self.distance13 = AngularDistance(self.star1, self.star3)
self.distance23 = AngularDistance(self.star2, self.star3)
distances.add_distance(self.star1, self.star3)
distances.add_distance(self.star1, self.star2)
distances.add_distance(self.star2, self.star3)
self.distances = distances
def test_find_close_several(self):
distances = self.distances
d_check = AngularDistance(self.star1, Star(5, 0, sph=(np.pi / 3, 0)))
epsilon = np.pi / 3 + 0.1
close = distances.find_close(d_check, epsilon)
self.assertEquals(len(close.distances), 2)
self.assertIn(self.distance12, close.distances)
self.assertIn(self.distance23, close.distances)
def test_find_close_all(self):
distances = self.distances
d_check = AngularDistance(self.star1, self.star4)
epsilon = np.pi
close = distances.find_close(d_check, epsilon)
self.assertEquals(len(close.distances), 3)
self.assertIn(self.distance12, close.distances)
self.assertIn(self.distance23, close.distances)
self.assertIn(self.distance13, close.distances)
def test_reduce_to_one_distance_all_gone(self):
distances = self.distances
d = AngularDistance(self.star4, Star(5, 0, sph=(0, 0)))
new_distances = distances.reduce_to_one_distance(d)
self.assertEquals(new_distances.distances, [])
def test_find_close_not_close_enough(self):
distances = self.distances
d_check = AngularDistance(self.star1, self.star4)
epsilon = 0.0001
close = distances.find_close(d_check, epsilon)
self.assertEquals(close.distances, [])
def test_find_close_single(self):
distances = self.distances
d_check = AngularDistance(self.star1, self.star4)
epsilon = 0.01
close = distances.find_close(d_check, epsilon)
self.assertEquals(close.distances, [self.distance23])