From 0ca412b19354529909c3619cf552186aa03f8e75 Mon Sep 17 00:00:00 2001
From: Stavros Efthymiou <35475381+stavros11@users.noreply.github.com>
Date: Wed, 14 Aug 2024 16:29:11 +0400
Subject: [PATCH] fix: rfsoc test

---
 tests/test_instruments_rfsoc.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_instruments_rfsoc.py b/tests/test_instruments_rfsoc.py
index 9bf867aa5d..f20e65408f 100644
--- a/tests/test_instruments_rfsoc.py
+++ b/tests/test_instruments_rfsoc.py
@@ -525,8 +525,8 @@ def test_classify_shots(dummy_qrc):
     target_shots = np.array([1, 1, 0, 0, 0, 0, 0])
 
     assert (target_shots == shots).all()
-    with pytest.raises(ValueError):
-        instrument.classify_shots(i_val, q_val, qubit1)
+    shots = instrument.classify_shots(i_val, q_val, qubit1)
+    assert shots.shape == (7,)
 
 
 def test_merge_sweep_results(dummy_qrc):