def test_get_brightness_threshold(self): #for image "spot1.jpg" self.assertEqual( bsd.get_brightness_threshold(bsd.get_image(self.images[0])), 2) #for image "spot2.jpg" self.assertEqual( bsd.get_brightness_threshold(bsd.get_image(self.images[1])), 41) #for image "spot3.jpg" self.assertEqual( bsd.get_brightness_threshold(bsd.get_image(self.images[2])), 25) #for image "spot4.jpg" - this file or image doesn't exist self.assertEqual( bsd.get_brightness_threshold(bsd.get_image(self.images[3])), None)
def test_get_image_center(self): #for image "spot1.jpg" self.assertEqual(bsd.get_image_center(bsd.get_image(self.images[0])), (539.5, 539.5)) #for image "spot2.jpg" self.assertEqual(bsd.get_image_center(bsd.get_image(self.images[1])), (134.5, 134.5)) #for image "spot3.jpg" self.assertEqual(bsd.get_image_center(bsd.get_image(self.images[2])), (1079.5, 1079.5)) #for image "spot4.jpg" self.assertEqual(bsd.get_image_center(bsd.get_image(self.images[3])), None)
def test_get_bright_pixel_list(self): #for image "spot1.jpg" self.assertTrue( (457, 520) in bsd.get_bright_pixel_list(bsd.get_image(self.images[0]))) self.assertTrue( (230, 412 ) not in bsd.get_bright_pixel_list(bsd.get_image(self.images[0]))) #for image "spot2.jpg" self.assertTrue( (203, 53) in bsd.get_bright_pixel_list(bsd.get_image(self.images[1]))) self.assertTrue( (162, 141 ) not in bsd.get_bright_pixel_list(bsd.get_image(self.images[1]))) #for image "spot3.jpg" self.assertTrue( (1651, 1663) in bsd.get_bright_pixel_list(bsd.get_image(self.images[2]))) self.assertTrue( (102, 121 ) not in bsd.get_bright_pixel_list(bsd.get_image(self.images[2]))) #for image "spot4.jpg" - this file or image doesn't exist self.assertEqual( bsd.get_bright_pixel_list(bsd.get_image(self.images[3])), None) pass
def test_get_bright_spot(self): #for image "spot1.jpg" self.assertTrue((457, 520) in bsd.get_bright_spot( bsd.get_neighborhood_list( bsd.get_bright_pixel_list(bsd.get_image(self.images[0]))))) self.assertTrue((230, 412) not in bsd.get_bright_spot( bsd.get_neighborhood_list( bsd.get_bright_pixel_list(bsd.get_image(self.images[0]))))) #for image "spot2.jpg" self.assertTrue((203, 53) in bsd.get_bright_spot( bsd.get_neighborhood_list( bsd.get_bright_pixel_list(bsd.get_image(self.images[1]))))) self.assertTrue((162, 141) not in bsd.get_bright_spot( bsd.get_neighborhood_list( bsd.get_bright_pixel_list(bsd.get_image(self.images[1]))))) #for image "spot3.jpg" self.assertTrue((1651, 1663) in bsd.get_bright_spot( bsd.get_neighborhood_list( bsd.get_bright_pixel_list(bsd.get_image(self.images[2]))))) self.assertTrue((102, 121) not in bsd.get_bright_spot( bsd.get_neighborhood_list( bsd.get_bright_pixel_list(bsd.get_image(self.images[2]))))) self.assertEqual(bsd.get_bright_spot([]), [])
def test_get_image(self): #for image "spot1.jpg" #for coordinates = (457,520) self.assertAlmostEqual(bsd.get_image(self.images[0]).getpixel( self.coordinate_sets[0][0]), 12, delta=1) #for coordinates = (230,412) self.assertAlmostEqual(bsd.get_image(self.images[0]).getpixel( self.coordinate_sets[0][1]), 1, delta=1) #for coordinates = (463,520) self.assertAlmostEqual(bsd.get_image(self.images[0]).getpixel( self.coordinate_sets[0][2]), 11, delta=1) #for coordinates = (937,607) self.assertAlmostEqual(bsd.get_image(self.images[0]).getpixel( self.coordinate_sets[0][3]), 0, delta=1) #for image "spot2.jpg" #for coordinates = (162,141) self.assertAlmostEqual(bsd.get_image(self.images[1]).getpixel( self.coordinate_sets[1][0]), 2, delta=1) #for coordinates = (203,53) self.assertAlmostEqual(bsd.get_image(self.images[1]).getpixel( self.coordinate_sets[1][1]), 254, delta=1) #for coordinates = (227,62) self.assertAlmostEqual(bsd.get_image(self.images[1]).getpixel( self.coordinate_sets[1][2]), 6, delta=1) #for image "spot3.jpg" #for coordinates = (102,121) self.assertAlmostEqual(bsd.get_image(self.images[2]).getpixel( self.coordinate_sets[2][0]), 0, delta=1) #for coordinates = (448,276) self.assertAlmostEqual(bsd.get_image(self.images[2]).getpixel( self.coordinate_sets[2][1]), 2, delta=1) #for coordinates = (1651,1663) self.assertAlmostEqual(bsd.get_image(self.images[2]).getpixel( self.coordinate_sets[2][2]), 238, delta=1) #for image "spot4.jpg" - this file or image doesn't exist self.assertEqual(bsd.get_image(self.images[3]), None) #for image "spot5.jpg" - there's a file with this name, but it's not an image self.assertEqual(bsd.get_image(self.images[4]), None)
from Tkinter import Tk from tkFileDialog import askopenfilename import bright_spot_distance as bsd #to avoid opening the main window --> Tk().withdraw() path = askopenfilename() image = bsd.get_image(path) bright_spot = bsd.get_bright_spot( bsd.get_neighborhood_list(bsd.get_bright_pixel_list(image))) distance_to_center = bsd.get_distance_to_center( bsd.get_image_center(image), bsd.get_bright_spot_center(bright_spot)) if (distance_to_center != None): print( "The distance between the bright spot and the center of the image is: {0:.2f}" .format(distance_to_center)) else: print("Invalid image or file!")