def test_optimize2(self, mock_init, mock_opt): """Test the case where the optimizer returns None (conflict).""" mock_db = Mock() mock_db.get_best.return_value = "(mock)", 1, 2 mock_db.get_result_count.return_value = 2 mock_model = Mock() mock_model.machine = MachineType() mock_model.fifos = [] mock_model.benchmarks = [MockBenchmark()] mock_init.return_value = Mock(), 10, 20 mock_opt(1, 2, 3, 4, 5).optimize.return_value = None mock_opt.reset_mock() optimize(mock_db, mock_model, 10, 5, ".") self.assertEqual(mock_init.call_count, 1) self.assertEqual(mock_db.get_best.call_count, 1) self.assertEqual(mock_opt.call_count, 1)
def test_optimize1(self, mock_init, mock_opt): """Test the typical case.""" result_count = [2] def mock_get_result_count(mod): temp = result_count[0] result_count[0] += 1 return temp mock_db = Mock() mock_db.get_best.return_value = "(mock)", 1, 2 mock_db.get_result_count.side_effect = mock_get_result_count mock_model = Mock() mock_model.machine = MachineType() mock_model.fifos = [] mock_model.benchmarks = [MockBenchmark()] mock_init.return_value = Mock(), 10, 20 optimize(mock_db, mock_model, 10, 5, ".") self.assertEqual(mock_init.call_count, 1) self.assertEqual(mock_db.get_best.call_count, 9) self.assertEqual(mock_opt.call_count, 1)