Commit b67e80f5 authored by Guillaume HEUSCH's avatar Guillaume HEUSCH

[unit test] fixed tests, removing unecessary image

parent 3d371094
Pipeline #22363 passed with stage
in 29 minutes and 11 seconds
......@@ -9,9 +9,6 @@
import numpy
import torch
from bob.io.base import load
from bob.io.base.test_utils import datafile
def test_architectures():
a = numpy.random.rand(1, 3, 128, 128).astype("float32")
......@@ -39,11 +36,10 @@ def test_architectures():
def test_transforms():
sample = {}
image = load(datafile('001.jpg', 'bob.learn.pytorch.test'))
image = numpy.random.rand(3, 128, 128).astype("uint8")
from ..datasets import RollChannels
sample['image'] = image
sample = {'image': image}
rc = RollChannels()
rc(sample)
assert sample['image'].shape == (128, 128, 3)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment