Completed
Push — master ( babbea...27f87c )
by Wojtek
9s
created

grortir.main.model.stages.CallsStage.get_cost()   A

Complexity

Conditions 1

Size

Total Lines 7

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 2
CRAP Score 1
Metric Value
cc 1
dl 0
loc 7
rs 9.4285
ccs 2
cts 2
cp 1
crap 1
1
"""Module contains class CallsStage."""
2 1
from grortir.main.model.core.abstract_stage import AbstractStage
3
4
5 1
class CallsStage(AbstractStage):
6
    """Implementation of basic stage.
7
8
    Cost is calculated by number of calls of cost function.
9
    """
10
11 1
    def __init__(self, input_vector=()):
12
        """Constructor."""
13 1
        super().__init__(input_vector)
14 1
        self.control_params = [0] * len(self.input_vector)
15 1
        self.cost = 0
16
17 1
    def get_quality(self):
0 ignored issues
show
Bug introduced by
Arguments number differs from overridden 'get_quality' method
Loading history...
18
        """
19
        Return quality of actual output.
20
21
        :return: quality
22
        """
23 1
        self.cost += 1
24 1
        return self.calculate_quality()
25
26 1
    def calculate_quality(self):
27
        """
28
        Function for calculating quality.
29
30
        :return:
31
        """
32 1
        assert len(self.control_params) == len(self.current_vector)
33 1
        quality = 0
34 1
        for i in enumerate(self.control_params):
35 1
            quality += (self.control_params[i[0]] - self.current_vector[
36
                i[0]]) ** 2
37 1
        return quality
38
39 1
    def get_cost(self):
0 ignored issues
show
Bug introduced by
Arguments number differs from overridden 'get_cost' method
Loading history...
40
        """
41
        Return actual cost of stage.
42
43
        :return: cost
44
        """
45
        return self.cost
46