def test_process_all_files_processess_all_files_group(self): groups = [[1, 2], [3, 4], [5, 6]] self.group_n_elements.return_value = groups process_all_files(self.files, self.remote_driver) expected_calls = [ call(groups[0], self.remote_driver), call(groups[1], self.remote_driver), call(groups[2], self.remote_driver) ] self.assertEqual(expected_calls, self.process_file_group.mock_calls)
def test_process_all_files_groups_n_elements(self): process_all_files(self.files, self.remote_driver) self.group_n_elements.assert_called_once_with(self.files, 20)