mindee-api-php/tests/V2/FileOperations/CropTest.php at 4933aa054fb7bb03b2ff5a5283e1609447458349 · mindee/mindee-api-php · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
<?php

declare(strict_types=1);

namespace V2\FileOperations;

use Mindee\Input\LocalResponse;
use Mindee\Input\PathInput;
use Mindee\V2\FileOperations\Crop;
use Mindee\V2\Product\Crop\CropResponse;
use PHPUnit\Framework\TestCase;
use TestingUtilities;

class CropTest extends TestCase
{
    private string $cropDataDir;

    protected function setUp(): void
    {
        $this->cropDataDir = TestingUtilities::getV2DataDir() . '/products/crop';
    }

    public function testProcessesSinglePageCropSplitCorrectly(): void
    {
        $inputSample = new PathInput($this->cropDataDir . '/default_sample.jpg');

        $localResponse = new LocalResponse($this->cropDataDir . '/crop_single.json');
        $doc = $localResponse->deserializeResponse(CropResponse::class);

        $cropOperation = new Crop($inputSample);
        $extractedCrops = $cropOperation->extractMultipleCrops($doc->inference->result->crops);

        self::assertCount(1, $extractedCrops);

        self::assertSame(0, $extractedCrops[0]->pageId);
        self::assertSame(0, $extractedCrops[0]->elementId);

        $bitmap0 = $extractedCrops[0]->image;

        self::assertSame(2822, $bitmap0->width ?? clone $bitmap0->getWidth());
        self::assertSame(1572, $bitmap0->height ?? clone $bitmap0->getHeight());
    }

    public function testProcessesMultiPageReceiptSplitCorrectly(): void
    {
        $inputSample = new PathInput($this->cropDataDir . '/multipage_sample.pdf');

        $localResponse = new LocalResponse($this->cropDataDir . '/crop_multiple.json');
        $doc = $localResponse->deserializeResponse(CropResponse::class);

        $cropOperation = new Crop($inputSample);
        $extractedCrops = $cropOperation->extractMultipleCrops($doc->inference->result->crops);

        self::assertCount(2, $extractedCrops);

        self::assertSame(0, $extractedCrops[0]->pageId);
        self::assertSame(0, $extractedCrops[0]->elementId);

        $bitmap0 = $extractedCrops[0]->image;
        self::assertSame(156, $bitmap0->width ?? $bitmap0->getWidth());
        self::assertSame(757, $bitmap0->height ?? $bitmap0->getHeight());

        self::assertSame(0, $extractedCrops[1]->pageId);
        self::assertSame(1, $extractedCrops[1]->elementId);

        $bitmap1 = $extractedCrops[1]->image;
        self::assertSame(188, $bitmap1->width ?? $bitmap1->getWidth());
        self::assertSame(691, $bitmap1->height ?? $bitmap1->getHeight());
    }
}