def test_change_image(self):
tool = AnnotationTool()
tool.load_app(True)
tool.file_list = []
tool.annotations = []
tool.current_file = 0
tool.class_list = ['winston', 'prince', 'duckie']
tool.colorspace = ['#0000FF', '#FF0000', '#00FF00']
tool.num_pages = 1
tool.class_count = [5, 5, 5]
for i in range(5):
a = Annotation()
a.rotation = 3
tool.file_list.append('file%d.jpg' % i)
for p in range(3):
roi = ROI()
roi.push(0,0)
roi.push(100.0,100.0)
a.push(roi, p%len(tool.class_list))
tool.annotations.append(a)
tool.img = MockImg()
tool._draw_workspace()
navigator = Navigator(tool)
self.assertEqual(tool.current_file, 0)
navigator.change_image(1)
self.assertEqual(tool.current_file, 1)
