Esempio n. 1
0
def test_is_batched_image():
    batched_image = torch.zeros(1, 1, 1, 1)
    assert image_.is_batched_image(batched_image)
    assert not image_.is_batched_image(batched_image.byte())
    assert not image_.is_batched_image(batched_image.squeeze(0))
Esempio n. 2
0
def test_make_batched_image():
    single_image = torch.empty(1, 1, 1)
    batched_image = image_.make_batched_image(single_image)
    assert image_.is_batched_image(batched_image)
Esempio n. 3
0
 def identity(batched_image):
     assert image_.is_batched_image(batched_image)
     return batched_image
Esempio n. 4
0
def test_read_image(test_image_file, test_image):
    actual = image_.read_image(test_image_file)
    desired = test_image
    assert image_.is_batched_image(actual)
    pyimagetest.assert_images_almost_equal(actual, desired)