# -*- coding: utf-8 -*-
import sys
from random import uniform
from unittest import TestCase
from PyQt5 import QtWidgets
from PyQt5.QtTest import QTest
from ....Classes.LamSlotWind import LamSlotWind
from ....Classes.MachineSCIM import MachineSCIM
from ....Classes.SlotW10 import SlotW10
from ....Classes.SlotW11 import SlotW11
from ....Classes.SlotW12 import SlotW12
from ....Classes.SlotW13 import SlotW13
from ....Classes.SlotW14 import SlotW14
from ....Classes.SlotW15 import SlotW15
from ....Classes.SlotW16 import SlotW16
from ....Classes.SlotW21 import SlotW21
from ....Classes.SlotW22 import SlotW22
from ....Classes.SlotW23 import SlotW23
from ....Classes.SlotW24 import SlotW24
from ....Classes.SlotW25 import SlotW25
from ....Classes.SlotW26 import SlotW26
from ....Classes.SlotW27 import SlotW27
from ....Classes.SlotW28 import SlotW28
from ....Classes.SlotW29 import SlotW29
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot10.PWSlot10 import PWSlot10
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot11.PWSlot11 import PWSlot11
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot12.PWSlot12 import PWSlot12
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot13.PWSlot13 import PWSlot13
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot14.PWSlot14 import PWSlot14
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot15.PWSlot15 import PWSlot15
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot16.PWSlot16 import PWSlot16
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot21.PWSlot21 import PWSlot21
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot22.PWSlot22 import PWSlot22
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot23.PWSlot23 import PWSlot23
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot24.PWSlot24 import PWSlot24
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot25.PWSlot25 import PWSlot25
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot26.PWSlot26 import PWSlot26
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot27.PWSlot27 import PWSlot27
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot28.PWSlot28 import PWSlot28
from ....GUI.Dialog.DMachineSetup.SWSlot.PWSlot29.PWSlot29 import PWSlot29
from ....GUI.Dialog.DMachineSetup.SWSlot.SWSlot import SWSlot
[docs]class test_SWSlot(TestCase):
"""Test that the widget SWSlot behave like it should"""
[docs] def setUp(self):
"""Run at the begining of every test to setup the gui"""
self.test_obj = MachineSCIM()
self.test_obj.stator = LamSlotWind(Rint=0.1, Rext=0.2)
self.test_obj.stator.slot = SlotW10(
Zs=123,
H0=0.10,
H1=0.11,
H2=0.12,
W0=0.13,
W1=0.14,
W2=0.15,
H1_is_rad=False,
)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
[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 slot"""
self.assertEqual(self.widget.si_Zs.value(), 123)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 10")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 0)
self.assertEqual(type(self.widget.w_slot), PWSlot10)
[docs] def test_set_slot_11(self):
"""Check that you can edit a Slot 11"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW11(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 11")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 1)
self.assertEqual(type(self.widget.w_slot), PWSlot11)
[docs] def test_set_slot_12(self):
"""Check that you can edit a Slot 12"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW12(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 12")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 2)
self.assertEqual(type(self.widget.w_slot), PWSlot12)
[docs] def test_set_slot_13(self):
"""Check that you can edit a Slot 13"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW13(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 13")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 3)
self.assertEqual(type(self.widget.w_slot), PWSlot13)
[docs] def test_set_slot_14(self):
"""Check that you can edit a Slot 14"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW14(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 14")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 4)
self.assertEqual(type(self.widget.w_slot), PWSlot14)
[docs] def test_set_slot_15(self):
"""Check that you can edit a Slot 15"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW15(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 15")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 5)
self.assertEqual(type(self.widget.w_slot), PWSlot15)
[docs] def test_set_slot_16(self):
"""Check that you can edit a Slot 16"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW16(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 16")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 6)
self.assertEqual(type(self.widget.w_slot), PWSlot16)
[docs] def test_set_slot_21(self):
"""Check that you can edit a Slot 21"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW21(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 21")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 7)
self.assertEqual(type(self.widget.w_slot), PWSlot21)
[docs] def test_set_slot_22(self):
"""Check that you can edit a Slot 22"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW22(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 22")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 8)
self.assertEqual(type(self.widget.w_slot), PWSlot22)
[docs] def test_set_slot_23(self):
"""Check that you can edit a Slot 23"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW23(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 23")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 9)
self.assertEqual(type(self.widget.w_slot), PWSlot23)
[docs] def test_set_slot_24(self):
"""Check that you can edit a Slot 24"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW24(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 24")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 10)
self.assertEqual(type(self.widget.w_slot), PWSlot24)
[docs] def test_set_slot_25(self):
"""Check that you can edit a Slot 25"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW25(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 25")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 11)
self.assertEqual(type(self.widget.w_slot), PWSlot25)
[docs] def test_set_slot_26(self):
"""Check that you can edit a Slot 26"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW26(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 26")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 12)
self.assertEqual(type(self.widget.w_slot), PWSlot26)
[docs] def test_set_slot_27(self):
"""Check that you can edit a Slot 27"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW27(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 27")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 13)
self.assertEqual(type(self.widget.w_slot), PWSlot27)
[docs] def test_set_slot_28(self):
"""Check that you can edit a Slot 28"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW28(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 28")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 14)
self.assertEqual(type(self.widget.w_slot), PWSlot28)
[docs] def test_set_slot_29(self):
"""Check that you can edit a Slot 29"""
Zs = int(uniform(1, 100))
self.test_obj.stator.slot = SlotW29(Zs=Zs)
self.widget = SWSlot(self.test_obj, matlib=[], is_stator=True)
self.assertEqual(self.widget.si_Zs.value(), Zs)
self.assertEqual(self.widget.c_slot_type.currentText(), "Slot Type 29")
self.assertEqual(self.widget.c_slot_type.currentIndex(), 15)
self.assertEqual(type(self.widget.w_slot), PWSlot29)
[docs] def test_set_Zs(self):
"""Check that the Widget allow to update Zs"""
# Clear the field before writing the new value
self.widget.si_Zs.clear()
value = int(uniform(1, 100))
QTest.keyClicks(self.widget.si_Zs, str(value))
self.widget.si_Zs.editingFinished.emit() # To trigger the slot
self.assertEqual(self.test_obj.stator.slot.Zs, value)
[docs] def test_c_slot_type(self):
"""Check that the combobox allow to update the slot type"""
slot_list = [
SlotW10,
SlotW11,
SlotW12,
SlotW13,
SlotW14,
SlotW15,
SlotW16,
SlotW21,
SlotW22,
SlotW23,
SlotW24,
SlotW25,
SlotW26,
SlotW27,
SlotW28,
SlotW29,
]
WIDGET_LIST = [
PWSlot10,
PWSlot11,
PWSlot12,
PWSlot13,
PWSlot14,
PWSlot15,
PWSlot16,
PWSlot21,
PWSlot22,
PWSlot23,
PWSlot24,
PWSlot25,
PWSlot26,
PWSlot27,
PWSlot28,
PWSlot29,
]
for ii in range(len(WIDGET_LIST)):
self.widget.c_slot_type.setCurrentIndex(ii)
self.assertEqual(type(self.test_obj.stator.slot), slot_list[ii])
self.assertEqual(type(self.widget.w_slot), WIDGET_LIST[ii])
self.assertEqual(
self.widget.w_slot.w_out.out_Wlam.text(), "Lamination width: 0.1 m"
)
