Refactored test_imaging_pipeline.py. For ticket #81