Source code for Tests.Plot.LamHole.test_Hole_58_plot

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

from os.path import join
import matplotlib.pyplot as plt
from numpy import pi

import pytest

from pyleecan.Classes.Frame import Frame
from pyleecan.Classes.LamHole import LamHole
from pyleecan.Classes.LamSlotWind import LamSlotWind
from pyleecan.Classes.MachineIPMSM import MachineIPMSM
from pyleecan.Classes.Magnet import Magnet
from pyleecan.Classes.Shaft import Shaft
from pyleecan.Classes.HoleM58 import HoleM58
from Tests import save_plot_path as save_path


class Test_Hole_58_plot(object):
    def test_Lam_Hole_58_plot(self):
        """Test machine plot hole 58"""

        plt.close("all")
        test_obj = LamHole(
            is_internal=True, Rint=0.021, Rext=0.075, is_stator=False, L1=0.7
        )
        test_obj.hole = list()
        test_obj.hole.append(
            HoleM58(
                Zh=8,
                W0=20e-3,
                W1=16e-3,
                W2=2e-3,
                W3=2 * pi / 8 * 0.6,
                H0=15e-3,
                H1=5e-3,
                H2=5e-3,
                R0=1e-3,
            )
        )

        test_obj.plot(is_show_fig=False)
        fig = plt.gcf()
        fig.savefig(join(save_path, "test_Lam_Hole_s58-Rotor.png"))
        assert len(fig.axes[0].patches) == 26

        test_obj.hole[0].plot(is_show_fig=False)
        fig = plt.gcf()
        fig.savefig(join(save_path, "test_Lam_Hole_s58-Rotor hole.png"))
        assert len(fig.axes[0].patches) == 3