예제 #1
0
파일: test_zz_lifio.py 프로젝트: jni/lesion
def test_series_iterator_limited():
    series_iter = lifio.series_iterator(test_lif, series=[0], c=0)
    assert isinstance(series_iter, coll.Iterator)
    series_list = list(series_iter)
    assert len(series_list) == 1
    assert series_list[0].shape == (1, 25, 1, 512, 512)
예제 #2
0
파일: test_zz_lifio.py 프로젝트: jni/lesion
def test_series_iterator():
    series_iter = lifio.series_iterator(test_lif)
    assert isinstance(series_iter, coll.Iterator)
    for series in series_iter:
        assert series.shape[-3:] == (4, 512, 512)