@@ -34,10 +34,11 @@ def test_gifti_image():
34
34
gi = GiftiImage ()
35
35
assert_equal (gi .numDA , 0 )
36
36
37
- da = GiftiDataArray (data = 'data' )
37
+ data = np .random .random ((5 ,))
38
+ da = GiftiDataArray .from_array (data )
38
39
gi .add_gifti_data_array (da )
39
40
assert_equal (gi .numDA , 1 )
40
- assert_equal (gi .darrays [0 ].data , ' data' )
41
+ assert_array_equal (gi .darrays [0 ].data , data )
41
42
42
43
gi .remove_gifti_data_array (0 )
43
44
assert_equal (gi .numDA , 0 )
@@ -49,14 +50,13 @@ def test_gifti_image():
49
50
50
51
# Remove one
51
52
gi = GiftiImage ()
52
- da = GiftiDataArray ( data = 'data' )
53
+ da = GiftiDataArray . from_array ( np . zeros (( 5 ,)), intent = 0 )
53
54
gi .add_gifti_data_array (da )
54
55
55
- gi .remove_gifti_data_array_by_intent (0 )
56
- assert_equal (gi .numDA , 1 )
56
+ gi .remove_gifti_data_array_by_intent (3 )
57
+ assert_equal (gi .numDA , 1 , "data array should exist on 'missed' remove" )
57
58
58
- gi .darrays [0 ].intent = 0
59
- gi .remove_gifti_data_array_by_intent (0 )
59
+ gi .remove_gifti_data_array_by_intent (da .intent )
60
60
assert_equal (gi .numDA , 0 )
61
61
62
62
0 commit comments