ENH: unit test for SplitSamplesProbabilityMapper and fix imports