Tests for imaging refactor

What it says
imaging tests