windpowerlib/tests/test_wind_turbine.py at 678f291d3a4c53ea4ef28dbbaf6991637bba5428 · wind-python/windpowerlib · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
"""
Testing the wind_turbine module.

SPDX-FileCopyrightText: 2019 oemof developer group <contact@oemof.org>
SPDX-License-Identifier: MIT
"""

import pytest
import os
from windpowerlib.tools import WindpowerlibUserWarning

from windpowerlib.wind_turbine import (get_turbine_data_from_file, WindTurbine,
                                       get_turbine_types, WindTurbineGroup,
                                       load_turbine_data_from_oedb)


class TestWindTurbine:

    @classmethod
    def setup_class(cls):
        """Setup default values"""
        cls.source = os.path.join(os.path.dirname(__file__), '../example/data')

    def test_warning(self, recwarn):
        test_turbine_data = {'hub_height': 100,
                             'rotor_diameter': 80,
                             'turbine_type': 'turbine_not_in_file',
                             'path': self.source}
        assert(WindTurbine(**test_turbine_data).power_curve is None)
        assert recwarn.pop(WindpowerlibUserWarning)

    def test_get_turbine_data_from_file(self):
        # Raise FileNotFoundError due to missing
        with pytest.raises(FileNotFoundError):
            get_turbine_data_from_file(turbine_type='...',
                                       path='not_existent')

    def test_get_turbine_types(self, capsys):
        get_turbine_types()
        captured = capsys.readouterr()
        assert 'Enercon' in captured.out
        get_turbine_types('oedb', print_out=False, filter_=False)
        msg = "`turbine_library` is 'wrong' but must be 'local' or 'oedb'."
        with pytest.raises(ValueError, match=msg):
            get_turbine_types('wrong')

    def test_wrong_url_load_turbine_data(self):
        """Load turbine data from oedb."""

        with pytest.raises(ConnectionError,
                           match="Database connection not successful"):
            load_turbine_data_from_oedb('wrong_schema')

    @pytest.mark.filterwarnings("ignore:The WindTurbine")
    def test_string_representation_of_wind_turbine(self):
        assert "Wind turbine: ['hub height=120 m'" in repr(WindTurbine(120))

    def test_power_curve_is_of_wrong_type(self):
        """Error raising due to wrong type of WindTurbine.power_curve."""
        test_turbine_data = {'hub_height': 100,
                             'rotor_diameter': 80,
                             'turbine_type': 'test_type',
                             'power_curve': 'string'}
        with pytest.raises(TypeError):
            WindTurbine(**test_turbine_data)

    def test_power_coefficient_curve_is_of_wrong_type(self):
        """Error raising due to wrong type of
        WindTurbine.power_coefficient_curve."""
        test_turbine_data = {'hub_height': 100,
                             'rotor_diameter': 80,
                             'turbine_type': 'test_type',
                             'power_coefficient_curve': 'string'}
        with pytest.raises(TypeError):
            WindTurbine(**test_turbine_data)

    def test_to_group_method(self):
        example_turbine = {
           'hub_height': 100,
           'rotor_diameter': 70,
           'turbine_type': 'DUMMY 3',
           'path': self.source}
        e_t_1 = WindTurbine(**example_turbine)
        assert(isinstance(e_t_1.to_group(), WindTurbineGroup))
        assert(e_t_1.to_group(5).number_of_turbines == 5)
        assert(e_t_1.to_group(number_turbines=5).number_of_turbines == 5)
        assert(e_t_1.to_group(total_capacity=3e6).number_of_turbines == 2.0)

    def test_wrongly_defined_to_group_method(self):
        example_turbine = {
           'hub_height': 100,
           'rotor_diameter': 70,
           'turbine_type': 'DUMMY 3',
           'path': self.source}
        e_t_1 = WindTurbine(**example_turbine)
        with pytest.raises(ValueError,
                           match="The 'number' and the 'total_capacity' "
                                 "parameter are mutually exclusive."):
            e_t_1.to_group(5, 3000)

    def test_create_unphysical_turbine(self):
        err_msg = "1/2rotor_diameter cannot be greater than hub_height"
        char = {
           'hub_height': 80,
           'rotor_diameter': 160,
           'turbine_type': 'DUMMY 3',
           'path': self.source
        }
        with pytest.raises(ValueError, match=err_msg):
            WindTurbine(**char)