Skip to content

Commit 017f7e5

Browse files
author
Ben Cipollini
committed
TST: Add tests for darrays / numDA
1 parent bc5f2d1 commit 017f7e5

File tree

1 file changed

+29
-0
lines changed

1 file changed

+29
-0
lines changed

nibabel/gifti/tests/test_gifti.py

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,35 @@ def test_gifti_image():
2626
gi = GiftiImage()
2727
assert_equal(gi.darrays, [])
2828

29+
# Test darrays / numDA
30+
gi = GiftiImage()
31+
assert_equal(gi.numDA, 0)
32+
33+
da = GiftiDataArray(data='data')
34+
gi.add_gifti_data_array(da)
35+
assert_equal(gi.numDA, 1)
36+
assert_equal(gi.darrays[0].data, 'data')
37+
38+
gi.remove_gifti_data_array(0)
39+
assert_equal(gi.numDA, 0)
40+
41+
# Remove from empty
42+
gi = GiftiImage()
43+
gi.remove_gifti_data_array_by_intent(0)
44+
assert_equal(gi.numDA, 0)
45+
46+
# Remove one
47+
gi = GiftiImage()
48+
da = GiftiDataArray(data='data')
49+
gi.add_gifti_data_array(da)
50+
51+
gi.remove_gifti_data_array_by_intent(0)
52+
assert_equal(gi.numDA, 1)
53+
54+
gi.darrays[0].intent = 0
55+
gi.remove_gifti_data_array_by_intent(0)
56+
assert_equal(gi.numDA, 0)
57+
2958

3059
def test_dataarray():
3160
for dt_code in data_type_codes.value_set():

0 commit comments

Comments
 (0)