Skip to content
Snippets Groups Projects
Commit b67e80f5 authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[unit test] fixed tests, removing unecessary image

parent 3d371094
No related branches found
No related tags found
1 merge request!3Resolve "Add unit tests"
Pipeline #
bob/learn/pytorch/test/data/001.jpg

2.23 KiB

......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment