<?php

use PHPUnit\Framework\TestCase;

use Pgvector\SparseVector;

final class SparseVectorTest extends TestCase
{
    public function testFromDense()
    {
        $embedding = new SparseVector([1, 0, 2, 0, 3, 0]);
        $this->assertEquals(6, $embedding->dimensions());
        $this->assertEquals([0, 2, 4], $embedding->indices());
        $this->assertEquals([1, 2, 3], $embedding->values());
    }

    public function testFromDenseSplFixedArray()
    {
        $embedding = new SparseVector(SplFixedArray::fromArray([1, 0, 2, 0, 3, 0]));
        $this->assertEquals('{1:1,3:2,5:3}/6', (string) $embedding);
    }

    public function testFromMap()
    {
        $map = [2 => 2, 4 => 3, 0 => 1, 3 => 0];
        $embedding = new SparseVector($map, 6);
        $this->assertEquals([1, 0, 2, 0, 3, 0], $embedding->toArray());
        $this->assertEquals([0, 2, 4], $embedding->indices());
        $this->assertEquals([2, 4, 0, 3], array_keys($map));
    }

    public function testFromString()
    {
        $embedding = new SparseVector('{1:1,3:2,5:3}/6');
        $this->assertEquals(6, $embedding->dimensions());
        $this->assertEquals([0, 2, 4], $embedding->indices());
        $this->assertEquals([1, 2, 3], $embedding->values());
    }

    public function testFromStringDimensions()
    {
        $this->expectException(InvalidArgumentException::class);
        $this->expectExceptionMessage('Extra argument');

        new SparseVector('{1:1,3:2,5:3}/6', 6);
    }

    public function testInvalidInteger()
    {
        $this->expectException(InvalidArgumentException::class);
        $this->expectExceptionMessage('Expected array');

        new SparseVector(1);
    }

    public function testToString()
    {
        $embedding = new SparseVector([1, 0, 2, 0, 3, 0]);
        $this->assertEquals('{1:1,3:2,5:3}/6', (string) $embedding);
    }

    public function testToArray()
    {
        $embedding = new SparseVector([1, 0, 2, 0, 3, 0]);
        $this->assertEquals([1, 0, 2, 0, 3, 0], $embedding->toArray());
    }
}