From 31bdc2e900e88fb205f89f56ea1d236844618f22 Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Mon, 6 Nov 2017 16:34:39 +0100
Subject: [PATCH] implement tests for parallel extractor training too
---
bob/bio/base/test/test_stacks.py | 12 +++++++++++-
1 file changed, 11 insertions(+), 1 deletion(-)
diff --git a/bob/bio/base/test/test_stacks.py b/bob/bio/base/test/test_stacks.py
index b1abac3c..cd6e0f53 100644
--- a/bob/bio/base/test/test_stacks.py
+++ b/bob/bio/base/test/test_stacks.py
@@ -47,7 +47,7 @@ def test_extractors():
assert all(np.allclose(x1, x2) for x1, x2 in zip(data, PAR_DATA))
-def test_trainable_extractors():
+def test_sequential_trainable_extractors():
processors = [CallableExtractor(p) for p in PROCESSORS] + [dummy_extractor]
proc = SequentialExtractor(processors)
with tempfile.NamedTemporaryFile(suffix='.hdf5') as f:
@@ -55,3 +55,13 @@ def test_trainable_extractors():
proc.load(f.name)
data = proc(DATA)
assert np.allclose(data, SEQ_DATA)
+
+
+def test_parallel_trainable_extractors():
+ processors = [CallableExtractor(p) for p in PROCESSORS] + [dummy_extractor]
+ proc = ParallelExtractor(processors)
+ with tempfile.NamedTemporaryFile(suffix='.hdf5') as f:
+ proc.train(DATA, f.name)
+ proc.load(f.name)
+ data = proc(np.array(DATA))
+ assert all(np.allclose(x1, x2) for x1, x2 in zip(data, PAR_DATA))
--
GitLab