Source code for pyleecan.Tests.Methods.Slot.test_HoleM54_meth

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

from unittest import TestCase
from ddt import ddt, data

from ....Classes.Segment import Segment
from ....Classes.SurfLine import SurfLine
from ....Classes.Arc1 import Arc1
from ....Classes.Arc3 import Arc3
from ....Classes.LamHole import LamHole
from ....Classes.HoleM54 import HoleM54
from numpy import exp, arcsin, ndarray, pi

# For AlmostEqual
DELTA = 1e-6

HoleM54_test = list()

test_obj = LamHole(is_internal=True, is_stator=False, hole=list(), Rext=0.1)
test_obj.hole = list()
test_obj.hole.append(HoleM54(Zh=8, W0=pi / 8, H0=30e-3, H1=10e-3, R1=60e-3))
HoleM54_test.append(
    {"test_obj": test_obj, "S_exp": 2.434734e-3, "Rmin": 0.06, "Rmax": 0.0724515}
)


@ddt
class test_HoleM54_meth(TestCase):
    """unittest for HoleM54 methods"""

    @data(*HoleM54_test)
    def test_comp_surface(self, test_dict):
        """Check that the computation of the surface is correct
        """
        test_obj = test_dict["test_obj"]
        result = test_obj.hole[0].comp_surface()

        a = result
        b = test_dict["S_exp"]
        msg = "Return " + str(a) + " expected " + str(b)
        self.assertAlmostEqual((a - b) / a, 0, delta=DELTA, msg=msg)

    @data(*HoleM54_test)
    def test_comp_radius(self, test_dict):
        """Check that the computation of the radius is correct
        """
        test_obj = test_dict["test_obj"]
        result = test_obj.hole[0].comp_radius()

        a = result[0]
        b = test_dict["Rmin"]
        msg = "For Rmin: Return " + str(a) + " expected " + str(b)
        self.assertAlmostEqual((a - b) / a, 0, delta=DELTA, msg=msg)

        a = result[1]
        b = test_dict["Rmax"]
        msg = "For Rmax: Return " + str(a) + " expected " + str(b)
        self.assertAlmostEqual((a - b) / a, 0, delta=DELTA, msg=msg)

    @data(*HoleM54_test)
    def test_build_geometry(self, test_dict):
        """Check that the surf list is correct
        """
        test_obj = LamHole(init_dict=test_dict["test_obj"].as_dict())
        result = test_obj.hole[0].build_geometry()

        self.assertEqual(len(result), 1)
        for surf in result:
            self.assertTrue(type(surf) == SurfLine)

        self.assertEqual(result[0].label[:4], "Hole")
        self.assertEqual(result[0].label[-9:], "_R0_T0_S0")
        self.assertEqual(len(result[0].line_list), 4)
        self.assertEqual(type(result[0].line_list[0]), Arc1)
        self.assertEqual(type(result[0].line_list[1]), Arc3)
        self.assertEqual(type(result[0].line_list[2]), Arc1)
        self.assertEqual(type(result[0].line_list[3]), Arc3)