TST: move the tests into separate files