Exemplo n.º 1
0
    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)
Exemplo n.º 2
0
    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)