# -*- coding: utf-8 -*-
import sys
from unittest import TestCase
from PyQt5 import QtWidgets
from ....Classes.Lamination import Lamination
from ....Classes.VentilationCirc import VentilationCirc
from ....Classes.VentilationPolar import VentilationPolar
from ....Classes.VentilationTrap import VentilationTrap
from ....GUI.Dialog.DMachineSetup.SLamParam.DAVDuct.DAVDuct import DAVDuct
from ....GUI.Dialog.DMachineSetup.SLamParam.DAVDuct.PVentCirc.PVentCirc import PVentCirc
from ....GUI.Dialog.DMachineSetup.SLamParam.DAVDuct.PVentPolar.PVentPolar import (
PVentPolar,
)
from ....GUI.Dialog.DMachineSetup.SLamParam.DAVDuct.PVentTrap.PVentTrap import PVentTrap
[docs]class test_DAVDuct(TestCase):
"""Test that the widget DAVDuct behave like it should"""
[docs] def setUp(self):
"""Run at the begining of every test to setup the gui"""
self.test_obj = Lamination(Rint=0.1, Rext=1, is_stator=True, is_internal=True)
self.test_obj.axial_vent.append(
VentilationCirc(Zh=8, H0=10e-3, D0=40e-3, Alpha0=0)
)
self.test_obj.axial_vent.append(
VentilationCirc(Zh=9, H0=20e-3, D0=50e-3, Alpha0=0)
)
self.widget = DAVDuct(self.test_obj)
[docs] @classmethod
def setUpClass(cls):
"""Start the app for the test"""
print("\nStart Test DAVDuct")
cls.app = QtWidgets.QApplication(sys.argv)
[docs] @classmethod
def tearDownClass(cls):
"""Exit the app after the test"""
cls.app.quit()
[docs] def test_init_circ(self):
"""Check that the Widget initialise for circular ventilations"""
self.assertEqual(self.widget.tab_vent.count(), 2)
self.assertEqual(self.widget.tab_vent.currentIndex(), 0)
# First set
self.assertEqual(type(self.widget.tab_vent.widget(0).w_vent), PVentCirc)
self.assertEqual(self.widget.tab_vent.widget(0).c_vent_type.currentIndex(), 0)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.si_Zh.value(), 8)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_H0.value(), 10e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_D0.value(), 40e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_Alpha0.value(), 0)
# Second set
self.assertEqual(type(self.widget.tab_vent.widget(1).w_vent), PVentCirc)
self.assertEqual(self.widget.tab_vent.widget(1).c_vent_type.currentIndex(), 0)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.si_Zh.value(), 9)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.lf_H0.value(), 20e-3)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.lf_D0.value(), 50e-3)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.lf_Alpha0.value(), 0)
[docs] def test_init_trap(self):
"""Check that the Widget initialise for polar ventilations"""
# Init the widget with Polar vent
self.test_obj.axial_vent = [
VentilationTrap(Zh=4, H0=24e-3, D0=34e-3, W1=44e-3, W2=54e-3, Alpha0=0)
]
self.widget = DAVDuct(self.test_obj)
self.assertEqual(self.widget.tab_vent.count(), 1)
self.assertEqual(self.widget.tab_vent.currentIndex(), 0)
# First set
self.assertEqual(type(self.widget.tab_vent.widget(0).w_vent), PVentTrap)
self.assertEqual(self.widget.tab_vent.widget(0).c_vent_type.currentIndex(), 1)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.si_Zh.value(), 4)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_H0.value(), 24e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_D0.value(), 34e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_W1.value(), 44e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_W2.value(), 54e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_Alpha0.value(), 0)
[docs] def test_init_polar(self):
"""Check that the Widget initialise for polar ventilations
"""
# Init the widget with Polar vent
vent = list()
vent.append(VentilationPolar(Zh=1, H0=21e-3, D0=31e-3, W1=41e-3, Alpha0=0))
vent.append(VentilationPolar(Zh=2, H0=22e-3, D0=32e-3, W1=42e-3, Alpha0=0))
vent.append(VentilationPolar(Zh=3, H0=23e-3, D0=33e-3, W1=43e-3, Alpha0=0))
self.test_obj.axial_vent = vent
self.widget = DAVDuct(self.test_obj)
self.assertEqual(self.widget.tab_vent.count(), 3)
self.assertEqual(self.widget.tab_vent.currentIndex(), 0)
# First set
self.assertEqual(type(self.widget.tab_vent.widget(0).w_vent), PVentPolar)
self.assertEqual(self.widget.tab_vent.widget(0).c_vent_type.currentIndex(), 2)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.si_Zh.value(), 1)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_H0.value(), 21e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_D0.value(), 31e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_W1.value(), 41e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_Alpha0.value(), 0)
# 2nd set
self.assertEqual(type(self.widget.tab_vent.widget(1).w_vent), PVentPolar)
self.assertEqual(self.widget.tab_vent.widget(1).c_vent_type.currentIndex(), 2)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.si_Zh.value(), 2)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.lf_H0.value(), 22e-3)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.lf_D0.value(), 32e-3)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.lf_W1.value(), 42e-3)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.lf_Alpha0.value(), 0)
# 3rd set
self.assertEqual(type(self.widget.tab_vent.widget(2).w_vent), PVentPolar)
self.assertEqual(self.widget.tab_vent.widget(2).c_vent_type.currentIndex(), 2)
self.assertEqual(self.widget.tab_vent.widget(2).w_vent.si_Zh.value(), 3)
self.assertEqual(self.widget.tab_vent.widget(2).w_vent.lf_H0.value(), 23e-3)
self.assertEqual(self.widget.tab_vent.widget(2).w_vent.lf_D0.value(), 33e-3)
self.assertEqual(self.widget.tab_vent.widget(2).w_vent.lf_W1.value(), 43e-3)
self.assertEqual(self.widget.tab_vent.widget(2).w_vent.lf_Alpha0.value(), 0)
[docs] def test_init_all_type(self):
"""Check that you can combine several kind of ventilations
"""
self.test_obj = Lamination(Rint=0.1, Rext=1, is_stator=True, is_internal=True)
self.test_obj.axial_vent.append(
VentilationCirc(Zh=8, H0=10e-3, D0=40e-3, Alpha0=0)
)
self.test_obj.axial_vent.append(
VentilationTrap(Zh=4, H0=24e-3, D0=34e-3, W1=44e-3, W2=54e-3, Alpha0=0)
)
self.test_obj.axial_vent.append(
VentilationPolar(Zh=3, H0=23e-3, D0=33e-3, W1=43e-3, Alpha0=0)
)
self.widget = DAVDuct(self.test_obj)
self.assertEqual(self.widget.tab_vent.count(), 3)
self.assertEqual(self.widget.tab_vent.currentIndex(), 0)
# First set
self.assertEqual(type(self.widget.tab_vent.widget(0).w_vent), PVentCirc)
self.assertEqual(self.widget.tab_vent.widget(0).c_vent_type.currentIndex(), 0)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.si_Zh.value(), 8)
# 2nd set
self.assertEqual(type(self.widget.tab_vent.widget(1).w_vent), PVentTrap)
self.assertEqual(self.widget.tab_vent.widget(1).c_vent_type.currentIndex(), 1)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.si_Zh.value(), 4)
# 3rd set
self.assertEqual(type(self.widget.tab_vent.widget(2).w_vent), PVentPolar)
self.assertEqual(self.widget.tab_vent.widget(2).c_vent_type.currentIndex(), 2)
self.assertEqual(self.widget.tab_vent.widget(2).w_vent.si_Zh.value(), 3)
[docs] def test_add_circ(self):
"""Test that you can add circ ventilation"""
self.widget.b_new.clicked.emit(True)
self.assertEqual(self.widget.tab_vent.count(), 3)
self.assertEqual(self.widget.tab_vent.currentIndex(), 0)
# First set
self.assertEqual(type(self.widget.tab_vent.widget(0).w_vent), PVentCirc)
self.assertEqual(self.widget.tab_vent.widget(0).c_vent_type.currentIndex(), 0)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.si_Zh.value(), 8)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_H0.value(), 10e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_D0.value(), 40e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_Alpha0.value(), 0)
# Second set
self.assertEqual(type(self.widget.tab_vent.widget(1).w_vent), PVentCirc)
self.assertEqual(self.widget.tab_vent.widget(1).c_vent_type.currentIndex(), 0)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.si_Zh.value(), 9)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.lf_H0.value(), 20e-3)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.lf_D0.value(), 50e-3)
self.assertEqual(self.widget.tab_vent.widget(1).w_vent.lf_Alpha0.value(), 0)
# Third set
self.assertEqual(type(self.widget.tab_vent.widget(2).w_vent), PVentCirc)
self.assertEqual(self.widget.tab_vent.widget(2).c_vent_type.currentIndex(), 0)
self.assertEqual(self.widget.tab_vent.widget(2).w_vent.si_Zh.value(), 8)
self.assertEqual(self.widget.tab_vent.widget(2).w_vent.lf_H0.value(), None)
self.assertEqual(self.widget.tab_vent.widget(2).w_vent.lf_D0.value(), None)
self.assertEqual(self.widget.tab_vent.widget(2).w_vent.lf_Alpha0.value(), None)
[docs] def test_remove_circ(self):
"""Test that you can remove circ ventilation"""
self.widget.b_remove.clicked.emit(True)
self.assertEqual(self.widget.tab_vent.count(), 1)
self.assertEqual(self.widget.tab_vent.currentIndex(), 0)
# First set
self.assertEqual(type(self.widget.tab_vent.widget(0).w_vent), PVentCirc)
self.assertEqual(self.widget.tab_vent.widget(0).c_vent_type.currentIndex(), 0)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.si_Zh.value(), 8)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_H0.value(), 10e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_D0.value(), 40e-3)
self.assertEqual(self.widget.tab_vent.widget(0).w_vent.lf_Alpha0.value(), 0)
