def test_next_image_last_image(self, mock_view):
args = mock.Mock()
args.input_dir = None
model = Model(args)
controller = Controller(model, mock_view)
model.images = ["abc.png", "test.png", "ababaa.png"]
model.current_file = 2
controller.next_image()
self.assertEqual(2, model.current_file)
def test_next_image(self, mock_view, mock_load_image):
args = mock.Mock()
args.input_dir = None
model = Model(args)
controller = Controller(model, mock_view)
model.images = ["abc.png", "test.png", "ababaa.png"]
model.current_file = 1
controller.next_image()
mock_load_image.assert_called_with = model.images[2]
self.assertEqual(2, model.current_file)