# -*- coding: utf-8 -*-
import sys
from random import uniform
from unittest import TestCase
from PyQt5 import QtWidgets
from PyQt5.QtTest import QTest
from ....Classes.Frame import Frame
from ....Classes.Shaft import Shaft
from ....Classes.LamSlotWind import LamSlotWind
from ....Classes.MachineSCIM import MachineSCIM
from ....GUI.Dialog.DMachineSetup.SMachineDimension.SMachineDimension import (
SMachineDimension,
)
[docs]class test_SMachineDimension(TestCase):
"""Test that the widget SMachineDimension 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(
is_stator=True, is_internal=False, Rint=0.21, Rext=0.22
)
self.test_obj.rotor = LamSlotWind(
is_stator=False, is_internal=True, Rint=0.11, Rext=0.12
)
self.test_obj.frame = Frame(Rint=0.22, Rext=0.24, Lfra=0.25)
self.test_obj.shaft = Shaft(Lshaft=0.333, Drsh=self.test_obj.rotor.Rint * 2)
self.widget = SMachineDimension(
machine=self.test_obj, matlib=[], is_stator=False
)
[docs] @classmethod
def setUpClass(cls):
"""Start the app for the test"""
print("\nStart Test SMachineDimension")
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 spinbox initialise to the lamination value"""
self.assertEqual(self.widget.lf_SRint.value(), 0.21)
self.assertEqual(self.widget.lf_SRext.value(), 0.22)
self.assertEqual(self.widget.lf_RRint.value(), 0.11)
self.assertEqual(self.widget.lf_RRext.value(), 0.12)
self.assertAlmostEqual(self.widget.lf_Wfra.value(), 0.02)
self.assertEqual(self.widget.lf_Lfra.value(), 0.25)
self.assertEqual(self.widget.g_frame.isChecked(), True)
self.assertEqual(self.widget.g_shaft.isChecked(), True)
self.assertEqual(self.widget.out_Drsh.text(), "Drsh = 2*Rotor.Rint = 220.0 mm")
[docs] def test_init_no_shaft(self):
"""Check that the Widget spinbox initialise to the lamination value"""
self.test_obj.shaft = None
self.widget = SMachineDimension(
machine=self.test_obj, matlib=[], is_stator=False
)
self.assertEqual(self.widget.g_shaft.isChecked(), False)
self.test_obj.shaft = Shaft(Drsh=None)
self.widget = SMachineDimension(
machine=self.test_obj, matlib=[], is_stator=False
)
self.assertEqual(self.widget.g_shaft.isChecked(), False)
self.test_obj.shaft = Shaft(Drsh=0)
self.widget = SMachineDimension(
machine=self.test_obj, matlib=[], is_stator=False
)
self.assertEqual(self.widget.g_shaft.isChecked(), False)
[docs] def test_set_SRint(self):
"""Check that the Widget allow to update SRint"""
# Clear the field before writing the new value
self.widget.lf_SRint.clear()
value = round(uniform(0, 1), 4)
QTest.keyClicks(self.widget.lf_SRint, str(value))
self.widget.lf_SRint.editingFinished.emit() # To trigger the slot
self.assertEqual(self.test_obj.stator.Rint, value)
[docs] def test_set_SRext(self):
"""Check that the Widget allow to update SRext"""
# Clear the field before writing the new value
self.widget.lf_SRext.clear()
value = round(uniform(0, 1), 4)
QTest.keyClicks(self.widget.lf_SRext, str(value))
self.widget.lf_SRext.editingFinished.emit() # To trigger the slot
self.assertEqual(self.test_obj.stator.Rext, value)
[docs] def test_set_RRint(self):
"""Check that the Widget allow to update RRint"""
# Clear the field before writing the new value
self.widget.lf_RRint.clear()
value = round(uniform(0, 1), 4)
QTest.keyClicks(self.widget.lf_RRint, str(value))
self.widget.lf_RRint.editingFinished.emit() # To trigger the slot
self.assertEqual(self.test_obj.rotor.Rint, value)
[docs] def test_set_RRext(self):
"""Check that the Widget allow to update RRext"""
# Clear the field before writing the new value
self.widget.lf_RRext.clear()
value = round(uniform(0, 1), 4)
QTest.keyClicks(self.widget.lf_RRext, str(value))
self.widget.lf_RRext.editingFinished.emit() # To trigger the slot
self.assertEqual(self.test_obj.rotor.Rext, value)
[docs] def test_set_Wfra(self):
"""Check that the Widget allow to update Wfra"""
# Clear the field before writing the new value
self.widget.lf_Wfra.clear()
value = round(uniform(0, 1), 4)
QTest.keyClicks(self.widget.lf_Wfra, str(value))
self.widget.lf_Wfra.editingFinished.emit() # To trigger the slot
self.assertEqual(self.test_obj.frame.Rint, self.test_obj.stator.Rext)
self.assertEqual(self.test_obj.frame.Rext, self.test_obj.stator.Rext + value)
[docs] def test_set_Lfra(self):
"""Check that the Widget allow to update Lfra"""
# Clear the field before writing the new value
self.widget.lf_Lfra.clear()
value = round(uniform(0, 1), 4)
QTest.keyClicks(self.widget.lf_Lfra, str(value))
self.widget.lf_Lfra.editingFinished.emit() # To trigger the slot
self.assertEqual(self.test_obj.frame.Lfra, value)
