# -*- coding: utf-8 -*-
import sys
from unittest import TestCase
from PyQt5 import QtWidgets
from ....Classes.LamHole import LamHole
from ....Classes.LamSlotWind import LamSlotWind
from ....Classes.MachineIPMSM import MachineIPMSM
from ....Classes.MachineSyRM import MachineSyRM
from ....Classes.Magnet import Magnet
from ....Classes.HoleM50 import HoleM50
from ....Classes.HoleM51 import HoleM51
from ....Classes.HoleM52 import HoleM52
from ....Classes.HoleM53 import HoleM53
from ....Classes.HoleM54 import HoleM54
from ....GUI.Dialog.DMachineSetup.SMHoleMag.SMHoleMag import SMHoleMag
from ....Classes.Material import Material
[docs]class test_SMHoleMag(TestCase):
"""Test that the widget SMHoleMag behave like it should"""
[docs] def setUp(self):
"""Run at the begining of every test to setup the gui"""
self.test_obj = MachineIPMSM(type_machine=8)
self.test_obj.stator = LamSlotWind()
self.test_obj.stator.winding.p = 4
self.test_obj.rotor = LamHole(Rint=0.1, Rext=0.2)
self.test_obj.rotor.hole = list()
self.test_obj.rotor.hole.append(HoleM50(Zh=8))
self.test_obj.rotor.hole[0].magnet_0.mat_type.name = "Magnet3"
self.test_obj2 = MachineSyRM(type_machine=5)
self.test_obj2.stator = LamSlotWind()
self.test_obj2.stator.winding.p = 4
self.test_obj2.rotor = LamHole(Rint=0.1, Rext=0.2)
self.test_obj2.rotor.hole = list()
self.test_obj2.rotor.hole.append(HoleM54(Zh=16))
self.matlib = list()
self.matlib.append(Material(name="Magnet1"))
self.matlib.append(Material(name="Magnet2"))
self.matlib.append(Material(name="Magnet3"))
self.widget = SMHoleMag(
machine=self.test_obj, matlib=self.matlib, is_stator=False
)
self.widget2 = SMHoleMag(
machine=self.test_obj2, matlib=self.matlib, is_stator=False
)
[docs] @classmethod
def setUpClass(cls):
"""Start the app for the test"""
print("\nStart Test SWSlot")
cls.app = QtWidgets.QApplication(sys.argv)
[docs] @classmethod
def tearDownClass(cls):
"""Exit the app after the test"""
cls.app.quit()
[docs] def test_init(self):
"""Check that the Widget initialize to the correct hole"""
self.assertEqual(
self.widget.out_hole_pitch.text(), "Slot pitch = 360 / 2p = 45 °"
)
self.assertEqual(self.widget.tab_hole.count(), 1)
self.assertEqual(self.widget.tab_hole.widget(0).c_hole_type.currentIndex(), 0)
self.assertEqual(
self.widget.tab_hole.widget(0).c_hole_type.currentText(), "Slot Type 50"
)
self.assertEqual(self.widget.tab_hole.widget(0).c_hole_type.count(), 4)
[docs] def test_init_SyRM(self):
"""Check that the Widget initialize to the correct hole"""
self.assertEqual(
self.widget2.out_hole_pitch.text(), "Slot pitch = 360 / 2p = 22.5 °"
)
self.assertEqual(self.widget2.tab_hole.count(), 1)
self.assertEqual(self.widget2.tab_hole.widget(0).c_hole_type.currentIndex(), 4)
self.assertEqual(
self.widget2.tab_hole.widget(0).c_hole_type.currentText(), "Slot Type 54"
)
self.assertEqual(self.widget2.tab_hole.widget(0).c_hole_type.count(), 5)
[docs] def test_init_SyRM_51(self):
"""Check that the Widget initialize to the correct hole"""
self.test_obj2.rotor.hole = list()
self.test_obj2.rotor.hole.append(
HoleM51(
Zh=16,
W0=0.11,
W1=0.12,
W2=0.13,
W3=0.14,
W4=0.15,
W5=0.16,
W6=0.17,
W7=0.18,
H0=0.19,
H1=0.2,
H2=0.21,
)
)
self.test_obj2.rotor.hole[0].remove_magnet()
self.widget2 = SMHoleMag(
machine=self.test_obj2, matlib=self.matlib, is_stator=False
)
self.assertEqual(
self.widget2.out_hole_pitch.text(), "Slot pitch = 360 / 2p = 22.5 °"
)
self.assertEqual(self.widget2.tab_hole.count(), 1)
self.assertEqual(self.widget2.tab_hole.widget(0).c_hole_type.currentIndex(), 1)
self.assertEqual(
self.widget2.tab_hole.widget(0).c_hole_type.currentText(), "Slot Type 51"
)
self.assertEqual(self.widget2.tab_hole.widget(0).c_hole_type.count(), 5)
self.assertEqual(self.widget2.tab_hole.widget(0).w_hole.lf_W0.text(), "0.11")
self.assertEqual(self.widget2.tab_hole.widget(0).w_hole.lf_W1.text(), "0.12")
self.assertEqual(self.widget2.tab_hole.widget(0).w_hole.lf_W2.text(), "0")
self.assertEqual(self.widget2.tab_hole.widget(0).w_hole.lf_W3.text(), "0")
self.assertEqual(self.widget2.tab_hole.widget(0).w_hole.lf_W4.text(), "0")
self.assertEqual(self.widget2.tab_hole.widget(0).w_hole.lf_W5.text(), "0")
self.assertEqual(self.widget2.tab_hole.widget(0).w_hole.lf_W6.text(), "0")
self.assertEqual(self.widget2.tab_hole.widget(0).w_hole.lf_W7.text(), "0")
self.assertEqual(self.widget2.tab_hole.widget(0).w_hole.lf_H0.text(), "0.19")
self.assertEqual(self.widget2.tab_hole.widget(0).w_hole.lf_H1.text(), "0.2")
self.assertEqual(self.widget2.tab_hole.widget(0).w_hole.lf_H2.text(), "0.21")
[docs] def test_init_51(self):
"""Check that you can edit a hole 51"""
self.test_obj.rotor.hole[0] = HoleM51(Zh=18)
self.test_obj.rotor.hole[0].magnet_0.mat_type.name = "Magnet1"
self.widget = SMHoleMag(
machine=self.test_obj, matlib=self.matlib, is_stator=False
)
self.assertEqual(
self.widget.out_hole_pitch.text(), "Slot pitch = 360 / 2p = 20 °"
)
self.assertEqual(self.widget.tab_hole.widget(0).c_hole_type.currentIndex(), 1)
self.assertEqual(
self.widget.tab_hole.widget(0).c_hole_type.currentText(), "Slot Type 51"
)
[docs] def test_init_52(self):
"""Check that you can edit a hole 52"""
self.test_obj.rotor.hole[0] = HoleM52(Zh=18)
self.test_obj.rotor.hole[0].magnet_0.mat_type.name = "Magnet1"
self.widget = SMHoleMag(
machine=self.test_obj, matlib=self.matlib, is_stator=False
)
self.assertEqual(
self.widget.out_hole_pitch.text(), "Slot pitch = 360 / 2p = 20 °"
)
self.assertEqual(self.widget.tab_hole.widget(0).c_hole_type.currentIndex(), 2)
self.assertEqual(
self.widget.tab_hole.widget(0).c_hole_type.currentText(), "Slot Type 52"
)
[docs] def test_init_53(self):
"""Check that you can edit a hole 53"""
self.test_obj.rotor.hole[0] = HoleM53(Zh=11)
self.test_obj.rotor.hole[0].magnet_0.mat_type.name = "Magnet1"
self.widget = SMHoleMag(
machine=self.test_obj, matlib=self.matlib, is_stator=False
)
self.assertEqual(
self.widget.out_hole_pitch.text(), "Slot pitch = 360 / 2p = 32.73 °"
)
self.assertEqual(self.widget.tab_hole.widget(0).c_hole_type.currentIndex(), 3)
self.assertEqual(
self.widget.tab_hole.widget(0).c_hole_type.currentText(), "Slot Type 53"
)
[docs] def test_set_type_51(self):
""" """
self.widget.tab_hole.widget(0).c_hole_type.setCurrentIndex(1)
self.assertEqual(type(self.test_obj.rotor.hole[0]), HoleM51)
self.assertEqual(
self.widget.out_hole_pitch.text(), "Slot pitch = 360 / 2p = 45 °"
)
self.assertEqual(self.widget.tab_hole.widget(0).c_hole_type.currentIndex(), 1)
self.assertEqual(
self.widget.tab_hole.widget(0).c_hole_type.currentText(), "Slot Type 51"
)
[docs] def test_set_type_52(self):
""" """
self.widget.tab_hole.widget(0).c_hole_type.setCurrentIndex(2)
self.assertEqual(type(self.test_obj.rotor.hole[0]), HoleM52)
self.assertEqual(
self.widget.out_hole_pitch.text(), "Slot pitch = 360 / 2p = 45 °"
)
self.assertEqual(self.widget.tab_hole.widget(0).c_hole_type.currentIndex(), 2)
self.assertEqual(
self.widget.tab_hole.widget(0).c_hole_type.currentText(), "Slot Type 52"
)
[docs] def test_set_type_53(self):
""" """
self.widget.tab_hole.widget(0).c_hole_type.setCurrentIndex(3)
self.assertEqual(type(self.test_obj.rotor.hole[0]), HoleM53)
self.assertEqual(
self.widget.out_hole_pitch.text(), "Slot pitch = 360 / 2p = 45 °"
)
self.assertEqual(self.widget.tab_hole.widget(0).c_hole_type.currentIndex(), 3)
self.assertEqual(
self.widget.tab_hole.widget(0).c_hole_type.currentText(), "Slot Type 53"
)
[docs] def test_SyRM_set_type_54_51_54(self):
"""Set a type 54 for a SyRM then set a 51 to check how the magnets are handled
"""
# Init a HoleM54
self.assertEqual(self.widget2.tab_hole.widget(0).c_hole_type.currentIndex(), 4)
self.assertEqual(
self.widget2.tab_hole.widget(0).c_hole_type.currentText(), "Slot Type 54"
)
self.assertEqual(self.test_obj2.rotor.hole[0].Zh, 16)
# Set type 51
self.widget2.tab_hole.widget(0).c_hole_type.setCurrentIndex(1)
self.assertEqual(self.widget2.tab_hole.widget(0).c_hole_type.currentIndex(), 1)
self.assertEqual(
self.widget2.tab_hole.widget(0).c_hole_type.currentText(), "Slot Type 51"
)
self.assertEqual(type(self.test_obj2.rotor.hole[0]), HoleM51)
self.assertEqual(self.test_obj2.rotor.hole[0].magnet_0, None)
self.assertEqual(self.test_obj2.rotor.hole[0].Zh, 16)
# Set type 54
self.widget2.tab_hole.widget(0).c_hole_type.setCurrentIndex(4)
self.assertEqual(self.widget2.tab_hole.widget(0).c_hole_type.currentIndex(), 4)
self.assertEqual(
self.widget2.tab_hole.widget(0).c_hole_type.currentText(), "Slot Type 54"
)
self.assertEqual(type(self.test_obj2.rotor.hole[0]), HoleM54)
self.assertEqual(self.test_obj2.rotor.hole[0].Zh, 16)
[docs] def test_add_remove_hole(self):
self.assertEqual(len(self.test_obj.rotor.hole), 1)
self.assertEqual(self.widget.tab_hole.count(), 1)
self.widget.b_add.clicked.emit(True)
self.assertEqual(len(self.test_obj.rotor.hole), 2)
self.assertEqual(type(self.test_obj.rotor.hole[1]), HoleM50)
self.assertEqual(self.test_obj.rotor.hole[1].Zh, 8)
self.assertEqual(self.widget.tab_hole.count(), 2)
self.widget.b_add.clicked.emit(True)
self.assertEqual(len(self.test_obj.rotor.hole), 3)
self.assertEqual(type(self.test_obj.rotor.hole[2]), HoleM50)
self.assertEqual(self.test_obj.rotor.hole[2].Zh, 8)
self.assertEqual(self.widget.tab_hole.count(), 3)
self.assertEqual(self.widget.tab_hole.tabText(0), "Slot 1")
self.assertEqual(self.widget.tab_hole.tabText(1), "Slot 2")
self.assertEqual(self.widget.tab_hole.tabText(2), "Slot 3")
self.widget.b_remove.clicked.emit(True)
self.assertEqual(len(self.test_obj.rotor.hole), 2)
self.assertEqual(type(self.test_obj.rotor.hole[1]), HoleM50)
self.assertEqual(self.widget.tab_hole.count(), 2)
self.widget.b_remove.clicked.emit(True)
self.assertEqual(len(self.test_obj.rotor.hole), 1)
self.assertEqual(self.widget.tab_hole.count(), 1)
# There is always at least 1 hole
self.widget.b_remove.clicked.emit(True)
self.assertEqual(len(self.test_obj.rotor.hole), 1)
self.assertEqual(self.widget.tab_hole.count(), 1)
[docs] def test_add_remove_hole_SyRM(self):
self.assertEqual(len(self.test_obj2.rotor.hole), 1)
self.assertEqual(self.widget2.tab_hole.count(), 1)
self.widget2.b_add.clicked.emit(True)
self.assertEqual(len(self.test_obj2.rotor.hole), 2)
self.assertEqual(type(self.test_obj2.rotor.hole[1]), HoleM50)
self.assertEqual(self.test_obj2.rotor.hole[1].Zh, 16)
self.assertEqual(self.test_obj2.rotor.hole[1].magnet_0, None)
self.assertEqual(self.widget2.tab_hole.count(), 2)
self.widget2.b_remove.clicked.emit(True)
self.assertEqual(len(self.test_obj2.rotor.hole), 1)
self.assertEqual(self.widget2.tab_hole.count(), 1)
# There is always at least 1 hole
self.widget2.b_remove.clicked.emit(True)
self.assertEqual(len(self.test_obj2.rotor.hole), 1)
self.assertEqual(self.widget2.tab_hole.count(), 1)
