Fix expected result of test

naplab · Dec 10, 2024 · 0edc81d · 0edc81d
1 parent 759ada9
commit 0edc81d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_data_object.py b/tests/test_data_object.py
@@ -58,7 +58,7 @@ def test_delete_trial_from_outstruct():
     data = {'x':[1,2], 'y':[3,4], 'z':[5,6]}
     outstruct = Data(data)
     del outstruct[0]
-    assert outstruct[0] == {'x':[2], 'y':[4], 'z':[6]}
+    assert outstruct[0] == {'x': 2, 'y': 4, 'z': 6}
 
 def test_create_outstruct_from_dict_different_lengths():
     data = {'x': [np.array([1,2]), np.array([3,4])], 'y': ['y0', 'y1', 'y2']}