def test_select_image(self):
tool = AnnotationTool()
tool.load_app(True)
appMenu = AppMenu(tool)
complete = appMenu.select_image()
self.assertTrue(complete)
self.assertTrue(hasattr(appMenu, 'prompt'))
self.assertTrue(hasattr(appMenu, 'prompt_entry'))
prompt_frame = appMenu.prompt.winfo_children()[0]
frame_child = prompt_frame.winfo_children()
arr = ["Move to Image #", '', "Ok", "Cancel"]
for i, child in enumerate(frame_child):
self.assertEqual(frame_child[i].cget('text'), arr[i])
def test_select_image_action(self):
tool = AnnotationTool()
tool.load_app(True)
tool.annotations = [Annotation(), Annotation(), Annotation()]
appMenu = AppMenu(tool)
appMenu.select_image()
appMenu.prompt_entry.insert(0, "1")
complete = appMenu.select_image_action()
self.assertTrue(complete)
appMenu.select_image()
appMenu.prompt_entry.insert(0, "4")
complete = appMenu.select_image_action()
self.assertFalse(complete)
appMenu.select_image()
appMenu.prompt_entry.insert(0, "-1")
complete = appMenu.select_image_action()
self.assertFalse(complete)
