MAINT: split up the tests into sampler tests and graph transform tests