From 4295fcc41e282f1f050caa71d350f584fc10bbff Mon Sep 17 00:00:00 2001
From: gifrerenom <lluis.gifre@cttc.es>
Date: Mon, 5 May 2025 16:23:43 +0000
Subject: [PATCH] Pre-merge code cleanup

---
 src/device/tests/test_unitary_morpheus.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/device/tests/test_unitary_morpheus.py b/src/device/tests/test_unitary_morpheus.py
index bb96289b6..6e4af77a2 100644
--- a/src/device/tests/test_unitary_morpheus.py
+++ b/src/device/tests/test_unitary_morpheus.py
@@ -76,9 +76,14 @@ def test_retrieve_state(driver: MorpheusApiDriver):
     state = driver.GetState(blocking=True)
 
     assert isinstance(state, types.GeneratorType), "Expected an iterator for state"
-    for i, item in enumerate(state):
-        #assert len(state) > 0, " State should not be empty"
-        print_data("State Item", item)
+    i = 0
+    for item in state:
+        if i == 0:
+            print_data("State Item", item)
+        else:
+            if item[1] == 'state': continue
+            print_data("Other Item", item)
+            i += 1
         if i > 3: break
 
     results = driver.UnsubscribeState([])
-- 
GitLab