moved unit tests to separate file:

because the mock module couldn't be found.