Source code for Tests.Methods.Slot.test_ventilationPolar

# -*- coding: utf-8 -*-

from os.path import join
import pytest
from numpy import pi

from pyleecan.Classes.VentilationPolar import VentilationPolar
from pyleecan.Methods.Slot.VentilationPolar import PolarArcBuildGeometryError


"""unittest for VentilationPolar"""


class Test_VentilationPolar(object):
    @pytest.fixture
    def vent(self):
        """Run at the begining of every test to setup the VentilationPolar"""
        test_obj = VentilationPolar(Zh=8, H0=0.08, D0=0.01, W1=pi / 8, Alpha0=pi / 8)

        return test_obj

    def test_build_geometry_errors(self, vent):
        """Test that build_geometry can raise some errors"""
        with pytest.raises(PolarArcBuildGeometryError) as context:
            vent.build_geometry(sym=0.2)
        with pytest.raises(PolarArcBuildGeometryError) as context:
            vent.build_geometry(sym=1, alpha="dz")
        with pytest.raises(PolarArcBuildGeometryError) as context:
            vent.build_geometry(sym=1, alpha=1, delta="dz")

    def test_build_geometry(self, vent):
        """Test that build_geometry works"""
        result = vent.build_geometry()

        assert len(result) == 8