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!")