Remove unittest

Author: ondras12345

tests/__init__.py

@@ -1 +0,0 @@
-__all__ = ['tests.test_OMParser', 'tests.test_ZMQ', 'tests.test_ModelicaSystem']

tests/test_ArrayDimension.py

@@ -1,34 +1,19 @@
 import OMPython
-import tempfile
-import shutil
-import os
 
 
-# do not change the prefix class name, the class name should have prefix "Test"
-# according to the documenation of pytest
-class Test_ArrayDimension:
-    def test_ArrayDimension(self):
-        omc = OMPython.OMCSessionZMQ()
+def test_ArrayDimension(tmp_path):
+    omc = OMPython.OMCSessionZMQ()
 
-        # create a temp dir for each session
-        tempdir = tempfile.mkdtemp()
-        if not os.path.exists(tempdir):
-            return print(tempdir, " cannot be created")
+    omc.sendExpression(f'cd("{tmp_path.as_posix()}")')
 
-        tempdirExp = "".join(["cd(", "\"", tempdir, "\"", ")"]).replace("\\", "/")
-        omc.sendExpression(tempdirExp)
+    omc.sendExpression('loadString("model A Integer x[5+1,1+6]; end A;")')
+    omc.sendExpression("getErrorString()")
 
-        omc.sendExpression("loadString(\"model A Integer x[5+1,1+6]; end A;\")")
-        omc.sendExpression("getErrorString()")
+    result = omc.sendExpression("getComponents(A)")
+    assert result[0][-1] == (6, 7), "array dimension does not match"
 
-        result = omc.sendExpression("getComponents(A)")
-        assert result[0][-1] == (6, 7), f"array dimension does not match the expected value. Got: {result[0][-1]}, Expected: {(6, 7)}"
+    omc.sendExpression('loadString("model A Integer y = 5; Integer x[y+1,1+9]; end A;")')
+    omc.sendExpression("getErrorString()")
 
-        omc.sendExpression("loadString(\"model A Integer y = 5; Integer x[y+1,1+9]; end A;\")")
-        omc.sendExpression("getErrorString()")
-
-        result = omc.sendExpression("getComponents(A)")
-        assert result[-1][-1] == ('y+1', 10), f"array dimension does not match the expected value. Got: {result[-1][-1]}, Expected: {('y+1', 10)}"
-
-        omc.__del__()
-        shutil.rmtree(tempdir, ignore_errors=True)
+    result = omc.sendExpression("getComponents(A)")
+    assert result[-1][-1] == ('y+1', 10), "array dimension does not match"

tests/test_FMIExport.py

@@ -1,34 +1,24 @@
 import OMPython
-import unittest
 import shutil
 import os
 
 
-class testFMIExport(unittest.TestCase):
-    def __init__(self, *args, **kwargs):
-        super(testFMIExport, self).__init__(*args, **kwargs)
-        self.tmp = ""
-
-    def __del__(self):
-        shutil.rmtree(self.tmp, ignore_errors=True)
-
-    def testCauerLowPassAnalog(self):
-        print("testing Cauer")
-        mod = OMPython.ModelicaSystem(modelName="Modelica.Electrical.Analog.Examples.CauerLowPassAnalog",
-                                      lmodel=["Modelica"])
-        self.tmp = mod.getWorkDirectory()
-
+def test_CauerLowPassAnalog():
+    mod = OMPython.ModelicaSystem(modelName="Modelica.Electrical.Analog.Examples.CauerLowPassAnalog",
+                                  lmodel=["Modelica"])
+    tmp = mod.getWorkDirectory()
+    try:
         fmu = mod.convertMo2Fmu(fileNamePrefix="CauerLowPassAnalog")
-        self.assertEqual(True, os.path.exists(fmu))
+        assert os.path.exists(fmu)
+    finally:
+        shutil.rmtree(tmp, ignore_errors=True)
 
-    def testDrumBoiler(self):
-        print("testing DrumBoiler")
-        mod = OMPython.ModelicaSystem(modelName="Modelica.Fluid.Examples.DrumBoiler.DrumBoiler", lmodel=["Modelica"])
-        self.tmp = mod.getWorkDirectory()
 
+def test_DrumBoiler():
+    mod = OMPython.ModelicaSystem(modelName="Modelica.Fluid.Examples.DrumBoiler.DrumBoiler", lmodel=["Modelica"])
+    tmp = mod.getWorkDirectory()
+    try:
         fmu = mod.convertMo2Fmu(fileNamePrefix="DrumBoiler")
-        self.assertEqual(True, os.path.exists(fmu))
-
-
-if __name__ == '__main__':
-    unittest.main()
+        assert os.path.exists(fmu)
+    finally:
+        shutil.rmtree(tmp, ignore_errors=True)

tests/test_FMIRegression.py

@@ -1,65 +1,68 @@
 import OMPython
 import tempfile
+import pathlib
 import shutil
 import os
 
 
-# do not change the prefix class name, the class name should have prefix "Test"
-# according to the documenation of pytest
-class Test_FMIRegression:
+def buildModelFMU(modelName):
+    omc = OMPython.OMCSessionZMQ()
 
-    def buildModelFMU(self, modelName):
-        omc = OMPython.OMCSessionZMQ()
-
-        # create a temp dir for each session
-        tempdir = tempfile.mkdtemp()
-        if not os.path.exists(tempdir):
-            return print(tempdir, " cannot be created")
-
-        tempdirExp = "".join(["cd(", "\"", tempdir, "\"", ")"]).replace("\\", "/")
-        omc.sendExpression(tempdirExp)
+    tempdir = pathlib.Path(tempfile.mkdtemp())
+    try:
+        omc.sendExpression(f'cd("{tempdir.as_posix()}")')
 
         omc.sendExpression("loadModel(Modelica)")
         omc.sendExpression("getErrorString()")
 
         fileNamePrefix = modelName.split(".")[-1]
-        exp = "buildModelFMU(" + modelName + ", fileNamePrefix=\"" + fileNamePrefix + "\"" + ")"
-
+        exp = f'buildModelFMU({modelName}, fileNamePrefix="{fileNamePrefix}")'
         fmu = omc.sendExpression(exp)
         assert os.path.exists(fmu)
-
-        omc.__del__()
+    finally:
+        del omc
         shutil.rmtree(tempdir, ignore_errors=True)
 
-    def test_Modelica_Blocks_Examples_Filter(self):
-        self.buildModelFMU("Modelica.Blocks.Examples.Filter")
 
-    def test_Modelica_Blocks_Examples_RealNetwork1(self):
-        self.buildModelFMU("Modelica.Blocks.Examples.RealNetwork1")
+def test_Modelica_Blocks_Examples_Filter():
+    buildModelFMU("Modelica.Blocks.Examples.Filter")
+
+
+def test_Modelica_Blocks_Examples_RealNetwork1():
+    buildModelFMU("Modelica.Blocks.Examples.RealNetwork1")
+
+
+def test_Modelica_Electrical_Analog_Examples_CauerLowPassAnalog():
+    buildModelFMU("Modelica.Electrical.Analog.Examples.CauerLowPassAnalog")
+
+
+def test_Modelica_Electrical_Digital_Examples_FlipFlop():
+    buildModelFMU("Modelica.Electrical.Digital.Examples.FlipFlop")
+
+
+def test_Modelica_Mechanics_Rotational_Examples_FirstGrounded():
+    buildModelFMU("Modelica.Mechanics.Rotational.Examples.FirstGrounded")
+
+
+def test_Modelica_Mechanics_Rotational_Examples_CoupledClutches():
+    buildModelFMU("Modelica.Mechanics.Rotational.Examples.CoupledClutches")
+
 
-    def test_Modelica_Electrical_Analog_Examples_CauerLowPassAnalog(self):
-        self.buildModelFMU("Modelica.Electrical.Analog.Examples.CauerLowPassAnalog")
+def test_Modelica_Mechanics_MultiBody_Examples_Elementary_DoublePendulum():
+    buildModelFMU("Modelica.Mechanics.MultiBody.Examples.Elementary.DoublePendulum")
 
-    def test_Modelica_Electrical_Digital_Examples_FlipFlop(self):
-        self.buildModelFMU("Modelica.Electrical.Digital.Examples.FlipFlop")
 
-    def test_Modelica_Mechanics_Rotational_Examples_FirstGrounded(self):
-        self.buildModelFMU("Modelica.Mechanics.Rotational.Examples.FirstGrounded")
+def test_Modelica_Mechanics_MultiBody_Examples_Elementary_FreeBody():
+    buildModelFMU("Modelica.Mechanics.MultiBody.Examples.Elementary.FreeBody")
 
-    def test_Modelica_Mechanics_Rotational_Examples_CoupledClutches(self):
-        self.buildModelFMU("Modelica.Mechanics.Rotational.Examples.CoupledClutches")
 
-    def test_Modelica_Mechanics_MultiBody_Examples_Elementary_DoublePendulum(self):
-        self.buildModelFMU("Modelica.Mechanics.MultiBody.Examples.Elementary.DoublePendulum")
+def test_Modelica_Fluid_Examples_PumpingSystem():
+    buildModelFMU("Modelica.Fluid.Examples.PumpingSystem")
 
-    def test_Modelica_Mechanics_MultiBody_Examples_Elementary_FreeBody(self):
-        self.buildModelFMU("Modelica.Mechanics.MultiBody.Examples.Elementary.FreeBody")
 
-    def test_Modelica_Fluid_Examples_PumpingSystem(self):
-        self.buildModelFMU("Modelica.Fluid.Examples.PumpingSystem")
+def test_Modelica_Fluid_Examples_TraceSubstances_RoomCO2WithControls():
+    buildModelFMU("Modelica.Fluid.Examples.TraceSubstances.RoomCO2WithControls")
 
-    def test_Modelica_Fluid_Examples_TraceSubstances_RoomCO2WithControls(self):
-        self.buildModelFMU("Modelica.Fluid.Examples.TraceSubstances.RoomCO2WithControls")
 
-    def test_Modelica_Clocked_Examples_SimpleControlledDrive_ClockedWithDiscreteTextbookController(self):
-        self.buildModelFMU("Modelica.Clocked.Examples.SimpleControlledDrive.ClockedWithDiscreteTextbookController")
+def test_Modelica_Clocked_Examples_SimpleControlledDrive_ClockedWithDiscreteTextbookController():
+    buildModelFMU("Modelica.Clocked.Examples.SimpleControlledDrive.ClockedWithDiscreteTextbookController")