From c179ebd8b31e3892fc2ddb7a203f916ea2228dcd Mon Sep 17 00:00:00 2001
From: ShreyaGursal <shreya.gursal@ansys.om>
Date: Thu, 26 Sep 2024 16:11:34 +0100
Subject: [PATCH] Adjustments made to the test.

---
 tests/test_lab.py | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/tests/test_lab.py b/tests/test_lab.py
index f7967e267..dc8de0820 100644
--- a/tests/test_lab.py
+++ b/tests/test_lab.py
@@ -19,12 +19,14 @@
 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
-import os
+
 from os import path, remove
+import time
 
 import pytest
 
 from RPC_Test_Common import get_dir_path, reset_to_default_file
+from ansys.motorcad.core import MotorCAD
 
 
 def test_model_build_lab(mc):
@@ -171,7 +173,10 @@ def test_external_custom_loss_functions(mc):
     assert mc.get_variable("NumCustomLossesExternal_Lab") == no_external_losses + 1
 
 
-def test_lab_model_export(mc):
+def test_lab_model_export():
+    mc = MotorCAD(open_new_instance=False)
+    mc.set_variable("MessageDisplayState", 2)
+    mc.load_template("e8")
     file_path = get_dir_path() + r"\test_files\temp_files\lab_model_export.lab"
 
     if path.exists(file_path):
@@ -181,6 +186,17 @@ def test_lab_model_export(mc):
 
     mc.export_lab_model(file_path)
 
+    # Exporting the lab model takes a few seconds and so a delay is required before
+    # asserting the .lab file is present.
+    checks = 0
+
+    while checks < 20:
+        time.sleep(1)
+        if path.exists(file_path) is False:
+            checks += 1
+        else:
+            break
+
     assert path.exists(file_path) is True
 
-    os.remove(file_path)
+    remove(file_path)