def test_run_lungmask_calls_apply_once_with_correct_parameters(
mock_lungmask_get_input_image, mock_lungmask_apply,
mock_lungmask_get_model, mock_time, mock_np_save,
mock_sitk_get_array_from_image):
def mock_exists(*args, **kwargs):
return True
os.path.exists = mock_exists
class MockInputImage():
def GetSpacing(self):
return (1, 2, 3)
class MockModel():
pass
mock_model = MockModel()
mock_input_image = MockInputImage()
mock_time.return_value = "123.123"
mock_lungmask_get_model.return_value = mock_model
mock_lungmask_get_input_image.return_value = mock_input_image
mock_model_name = "mock model name"
param_dict = {
"source_dir": ["/source/dir"],
"model_name": [mock_model_name]
}
run_lungmask(param_dict)
mock_lungmask_apply.assert_called_once_with(mock_input_image,
mock_model,
force_cpu=False,
batch_size=20,
volume_postprocessing=False)
def test_run_lungmask_calls_get_model_once_with_name_from_param_dict(
mock_lungmask_get_input_image, mock_lungmask_apply,
mock_lungmask_get_model, mock_time, mock_np_save,
mock_sitk_get_array_from_image):
def mock_exists(*args, **kwargs):
return True
os.path.exists = mock_exists
class MockInputImage():
def GetSpacing(self):
return (1, 2, 3)
mock_time.return_value = "123.123"
mock_lungmask_get_model.return_value = "MockModel123"
mock_lungmask_get_input_image.return_value = MockInputImage()
mock_model_name = "mock model name"
param_dict = {
"source_dir": ["/source/dir"],
"model_name": [mock_model_name]
}
run_lungmask(param_dict)
mock_lungmask_get_model.assert_called_once_with('unet', mock_model_name)
def test_run_lungmask_calls_numpy_save_twice(mock_lungmask_get_input_image,
mock_lungmask_apply,
mock_lungmask_get_model,
mock_time, mock_np_save,
mock_sitk_get_array_from_image):
def mock_exists(*args, **kwargs):
return True
os.path.exists = mock_exists
class MockInputImage():
def GetSpacing(self):
return (1, 2, 3)
class MockModel():
pass
mock_model = MockModel()
mock_input_image = MockInputImage()
mock_time.return_value = "123.123"
mock_lungmask_get_model.return_value = mock_model
mock_lungmask_get_input_image.return_value = mock_input_image
mock_model_name = "mock model name"
param_dict = {
"source_dir": ["/source/dir"],
"model_name": [mock_model_name]
}
run_lungmask(param_dict)
assert mock_np_save.call_count == 2
def test_run_lungmask_result_dictionary_contains_segmentation_and_input(
mock_lungmask_get_input_image, mock_lungmask_apply,
mock_lungmask_get_model, mock_time, mock_np_save,
mock_sitk_get_array_from_image):
def mock_exists(*args, **kwargs):
return True
os.path.exists = mock_exists
class MockInputImage():
def GetSpacing(self):
return (1, 2, 3)
class MockModel():
pass
mock_model = MockModel()
mock_input_image = MockInputImage()
mock_time.return_value = "123.123"
mock_lungmask_get_model.return_value = mock_model
mock_lungmask_get_input_image.return_value = mock_input_image
mock_model_name = "mock model name"
param_dict = {
"source_dir": ["/source/dir"],
"model_name": [mock_model_name]
}
result_dict, success = run_lungmask(param_dict)
assert success
assert "segmentation" in result_dict
assert "input" in result_dict
def test_run_lungmask_sends_success_false_when_input_image_doesnt_exist(
mock_lungmask_get_input_image, mock_lungmask_apply,
mock_lungmask_get_model, mock_time, mock_np_save,
mock_sitk_get_array_from_image):
def mock_exists_false(*args, **kwargs):
return False
os.path.exists = mock_exists_false
class MockInputImage():
def GetSpacing(self):
return (1, 2, 3)
class MockModel():
pass
mock_model = MockModel()
mock_input_image = MockInputImage()
mock_time.return_value = "123.123"
mock_lungmask_get_model.return_value = mock_model
mock_lungmask_get_input_image.return_value = mock_input_image
mock_model_name = "mock model name"
param_dict = {
"source_dir": ["/source/dir"],
"model_name": [mock_model_name]
}
result_dict, success = run_lungmask(param_dict)
assert not success