mmpose/tests/test_datasets/test_transforms/test_loading.py

# Copyright (c) OpenMMLab. All rights reserved.
from unittest import TestCase

import numpy as np
from mmcv import imread

from mmpose.datasets.transforms.loading import LoadImage


class TestLoadImage(TestCase):

    def test_load_image(self):

        transform = LoadImage()
        results = dict(img_path='tests/data/coco/000000000785.jpg')

        results = transform(results)

        self.assertIsInstance(results['img'], np.ndarray)

    def test_with_input_image(self):
        transform = LoadImage(to_float32=True)

        img_path = 'tests/data/coco/000000000785.jpg'
        results = dict(
            img_path=img_path, img=imread(img_path).astype(np.uint8))

        results = transform(results)

        self.assertIsInstance(results['img'], np.ndarray)
        self.assertTrue(results['img'].dtype, np.float32)