TestOptimizer

Complexity

Total Complexity

4

Size/Duplication

Total Lines	21
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	test_set_order()	0	5	1
A	test___init__()	0	5	1
A	test_set_order_fail()	0	6	2

"""Module for testing optimizer."""

from unittest import TestCase

from grortir.main.model.core.abstract_process import AbstractProcess
from grortir.main.model.core.abstract_stage import AbstractStage
from grortir.main.optimizers.optimizer import Optimizer

SOME_PROCESS = AbstractProcess()
FIRST_STAGE = AbstractStage()
SECOND_STAGE = AbstractStage()
SOME_PROCESS.add_path([FIRST_STAGE, SECOND_STAGE])

The Instance of AbstractProcess does not seem to have a member named add_path.

class TestOptimizer(TestCase):
    """Class for testing Optimizer."""

    def test___init__(self):
        """Testing creating object."""
        optimizer = Optimizer(SOME_PROCESS)
        self.assertIsNotNone(optimizer)
        self.assertEqual(optimizer.process, SOME_PROCESS)

    def test_set_order(self):
        """Test changing order of optimizing."""
        optimizer = Optimizer(SOME_PROCESS)
        optimizer.set_custom_optimizing_order([SECOND_STAGE, FIRST_STAGE])
        self.assertEqual(optimizer.ordered_stages[0], SECOND_STAGE)

    def test_set_order_fail(self):
        """Test changing order of optimizing with incorrect list."""
        optimizer = Optimizer(SOME_PROCESS)
        with self.assertRaises(ValueError):
            optimizer.set_custom_optimizing_order(
                [SECOND_STAGE, FIRST_STAGE, FIRST_STAGE])