def test_create_params():
preprocess_ct.Params()
with pytest.raises(TypeError):
preprocess_ct.Params(clip_lower='one', clip_upper=0)
preprocess_ct.Params(clip_lower=1, clip_upper=0)
preprocess_ct.Params(ndim=0)
preprocess_ct.Params(min_max_normalize=[False])
def test_preprocess_dicom_min_max_scale(dicom_path):
params = preprocess_ct.Params(clip_lower=-1000, clip_upper=400, min_max_normalize=True)
preprocess = preprocess_ct.PreprocessCT(params)
dicom_array, meta = load_ct.load_ct(dicom_path)
meta = load_ct.MetaData(meta)
dicom_array = preprocess(dicom_array, meta)
assert isinstance(dicom_array, np.ndarray)
assert dicom_array.max() <= 1
assert dicom_array.min() >= 0
def test_preprocess_dicom_clips(dicom_path):
params = preprocess_ct.Params(clip_lower=-1, clip_upper=40)
preprocess = preprocess_ct.PreprocessCT(params)
dicom_array, meta = load_ct.load_ct(dicom_path)
meta = load_ct.MetaData(meta)
dicom_array = preprocess(dicom_array, meta)
assert isinstance(dicom_array, np.ndarray)
assert dicom_array.max() <= 40
assert dicom_array.min() >= -1
def test_preprocess_dicom_pure(dicom_path):
params = preprocess_ct.Params()
preprocess = preprocess_ct.PreprocessCT(params)
dicom_array, meta = load_ct.load_dicom(dicom_path)
assert isinstance(dicom_array, np.ndarray)
dicom_array, meta = load_ct.load_dicom(dicom_path)
meta = load_ct.MetaData(meta)
dicom_array = preprocess(dicom_array, meta)
assert isinstance(dicom_array, np.ndarray)
def test_create_params():
preprocess_ct.Params()
params = preprocess_ct.Params(spacing=1., ndim=3)
assert len(params.spacing) == 3
spacing = [shape == 1. for shape in params.spacing]
assert all(spacing)
with pytest.raises(TypeError):
preprocess_ct.Params(clip_lower='one', clip_upper=0)
preprocess_ct.Params(clip_lower=1, clip_upper=0)
preprocess_ct.Params(ndim=0)
preprocess_ct.Params(min_max_normalize=[False])
with pytest.raises(RuntimeError):
preprocess_ct.Params(spacing=(1, 1, 1, 1), ndim=3)