<?php

declare (strict_types = 1);

namespace tests\Clustering;

use Phpml\Clustering\KMeans;

class KMeansTest extends \PHPUnit_Framework_TestCase
{
    public function testKMeansSamplesClustering()
    {
        $samples = [[1, 1], [8, 7], [1, 2], [7, 8], [2, 1], [8, 9]];

        $kmeans = new KMeans(2);
        $clusters = $kmeans->cluster($samples);

        $this->assertEquals(2, count($clusters));

        foreach ($samples as $index => $sample) {
            if (in_array($sample, $clusters[0]) || in_array($sample, $clusters[1])) {
                unset($samples[$index]);
            }
        }
        $this->assertEquals(0, count($samples));
    }

    public function testKMeansMoreSamplesClustering()
    {
        $samples = [
            [80, 55], [86, 59], [19, 85], [41, 47], [57, 58],
            [76, 22], [94, 60], [13, 93], [90, 48], [52, 54],
            [62, 46], [88, 44], [85, 24], [63, 14], [51, 40],
            [75, 31], [86, 62], [81, 95], [47, 22], [43, 95],
            [71, 19], [17, 65], [69, 21], [59, 60], [59, 12],
            [15, 22], [49, 93], [56, 35], [18, 20], [39, 59],
            [50, 15], [81, 36], [67, 62], [32, 15], [75, 65],
            [10, 47], [75, 18], [13, 45], [30, 62], [95, 79],
            [64, 11], [92, 14], [94, 49], [39, 13], [60, 68],
            [62, 10], [74, 44], [37, 42], [97, 60], [47, 73],
        ];

        $kmeans = new KMeans(4);
        $clusters = $kmeans->cluster($samples);

        $this->assertEquals(4, count($clusters));

        foreach ($samples as $index => $sample) {
            for ($i = 0; $i < 4; ++$i) {
                if (in_array($sample, $clusters[$i])) {
                    unset($samples[$index]);
                }
            }
        }
        $this->assertEquals(0, count($samples));
    }
}