def test_get_slide_name(self):
db = DBConnection("../data_base/test_base.db")
db.insert_slide("1","1.jpg")
db.insert_object("1","LOL")
disp = InfoDisplayer(db, "TestImages/")
name = disp.get_slide_name(1)
self.assertTrue(name == "1.jpg")
db.clear()
def test_insert_slide(self):
db = DBConnection("../data_base/test_base.db")
self.assertTrue(db.get_slide_count() == 0)
db.insert_slide(1, "1")
self.assertTrue(db.get_slide_count() == 1)
self.assertTrue(db.get_slide_name(1) == "1")
db.clear()
self.assertTrue(db.get_slide_count() == 0)
def test_delete_object(self):
db = DBConnection("../data_base/test_base.db")
self.assertTrue(db.get_object_count() == 0)
db.insert_object(1, "1")
self.assertTrue(db.get_object_count() == 1)
self.assertTrue(db.get_object_name(1) == "1")
db.delete_object(1)
self.assertTrue(db.get_object_count() == 0)
db.clear()
def test_insert_object(self):
db = DBConnection("../data_base/test_base.db")
db.clear()
self.assertRaises(DBConnection("RANDOM_BASE"))
self.assertTrue(db.get_object_count() == 0)
db.insert_object(1, "1")
self.assertTrue(db.get_object_count() == 1)
self.assertTrue(db.get_object_name(1) == "1")
db.clear()
self.assertTrue(db.get_object_count() == 0)
def test_display(self):
db = DBConnection("../data_base/test_base.db")
db.insert_slide("1","1.jpg")
db.insert_object("1","LOL")
disp = InfoDisplayer(db, "TestImages/")
im = cv2.imread("TestImages/1.jpg")
im2 = cv2.imread("TestImages/2.jpg")
im2 = disp.display(1,im2)
db.clear()
self.assertTrue(len(im) == len(im2))
for x in range(im.shape[0]):
for y in range(im.shape[1]):
self.assertTrue(im[x][y][0]==im2[x][y][0])
def test_get_first(self):
db = DBConnection("../data_base/test_base.db")
db.insert_slide("1","1.jpg")
db.insert_slide("2","2.jpg")
db.insert_object("1","LOL")
db.insert_object("2","LOL")
ob1 = (1,"LOL")
ob2 = (2,"LOL")
ob3 = (3,"LOL")
disp = InfoDisplayer(db, "TestImages/")
out = disp.get_first(ob1)
self.assertTrue(out == "1.jpg")
out = disp.get_first(ob2)
self.assertTrue(out == "2.jpg")
out = disp.get_first(ob3)
self.assertTrue(out == "")
db.clear()
def test_get_slide_list(self):
db = DBConnection("../data_base/test_base.db")
self.assertTrue(db.get_slide_count() == 0)
db.insert_slide(1, "1")
db.insert_slide(2, "2")
db.insert_slide(3, "3")
db.insert_slide(4, "4")
db.insert_slide(5, "5")
db.insert_slide(6, "6")
self.assertTrue(db.get_slide_count() == 6)
test_list = db.get_slide_list()
self.assertTrue(test_list[0][2] == "1")
self.assertTrue(test_list[1][2] == "2")
self.assertTrue(test_list[2][2] == "3")
self.assertTrue(test_list[3][2] == "4")
self.assertTrue(test_list[4][2] == "5")
self.assertTrue(test_list[5][2] == "6")
db.clear()
self.assertTrue(db.get_object_count() == 0)
