diff --git a/src/device/tests/test_unitary_morpheus.py b/src/device/tests/test_unitary_morpheus.py index bb96289b6b1685d8a56d9ce5b1ef096cb713591d..6e4af77a2d32a9d8c424a9ee98e2328e721c2aa9 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([])