public function testDatasetStratifiedRandomSplitWithEvenDistributionAndNumericTargets()
{
$dataset = new ArrayDataset($samples = [[1], [2], [3], [4], [5], [6], [7], [8]], $labels = [1, 2, 1, 2, 1, 2, 1, 2]);
$split = new StratifiedRandomSplit($dataset, 0.5);
$this->assertEquals(2, $this->countSamplesByTarget($split->getTestLabels(), 1));
$this->assertEquals(2, $this->countSamplesByTarget($split->getTestLabels(), 2));
$split = new StratifiedRandomSplit($dataset, 0.25);
$this->assertEquals(1, $this->countSamplesByTarget($split->getTestLabels(), 1));
$this->assertEquals(1, $this->countSamplesByTarget($split->getTestLabels(), 2));
}
