@@ -11,7 +11,7 @@ def test_ls_one():
1111 result = runner .invoke (cli .main , ['ls' , 'MNI152Lin' , '--res' , '1' , '-s' , 'T1w' ])
1212
1313 # One result
14- lines = result .output .strip ().splitlines ()
14+ lines = result .stdout .strip ().splitlines ()
1515 assert len (lines ) == 1
1616
1717 assert 'tpl-MNI152Lin/tpl-MNI152Lin_res-01_T1w.nii.gz' in lines [0 ]
@@ -27,7 +27,7 @@ def test_ls_multi():
2727 result = runner .invoke (cli .main , ['ls' , 'MNI152Lin' , '--res' , '1' , '-s' , 'T1w' , '-s' , 'T2w' ])
2828
2929 # Two results
30- lines = result .output .strip ().splitlines ()
30+ lines = result .stdout .strip ().splitlines ()
3131 assert len (lines ) == 2
3232
3333 assert 'tpl-MNI152Lin/tpl-MNI152Lin_res-01_T1w.nii.gz' in lines [0 ]
@@ -44,10 +44,7 @@ def test_get_one():
4444 result = runner .invoke (cli .main , ['get' , 'MNI152Lin' , '--res' , '1' , '-s' , 'T1w' ])
4545
4646 # One result, possible download status before
47- lines = result .output .strip ().splitlines ()[- 2 :]
48- if len (lines ) == 2 :
49- assert lines [0 ].startswith ('100%' )
50- lines .pop (0 )
47+ lines = result .stdout .strip ().splitlines ()[- 2 :]
5148
5249 assert 'tpl-MNI152Lin/tpl-MNI152Lin_res-01_T1w.nii.gz' in lines [0 ]
5350
@@ -63,10 +60,7 @@ def test_get_multi():
6360 result = runner .invoke (cli .main , ['get' , 'MNI152Lin' , '--res' , '1' , '-s' , 'T1w' , '-s' , 'T2w' ])
6461
6562 # Two result, possible download status before
66- lines = result .output .strip ().splitlines ()[- 3 :]
67- if len (lines ) == 3 :
68- assert lines [0 ].startswith ('100%' )
69- lines .pop (0 )
63+ lines = result .stdout .strip ().splitlines ()[- 3 :]
7064
7165 assert 'tpl-MNI152Lin/tpl-MNI152Lin_res-01_T1w.nii.gz' in lines [0 ]
7266 assert 'tpl-MNI152Lin/tpl-MNI152Lin_res-01_T2w.nii.gz' in lines [1 ]
0 commit comments