From 3dd816daf17412cf0bfc9b0bf5aa0c59aac731aa Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Mon, 29 Mar 2021 18:42:27 +0200
Subject: [PATCH] [tests] Mark tensorflow tests as slow related to #43

---
 bob/bio/face/test/test_baselines.py    | 7 +++++++
 bob/bio/face/test/test_embeddings.py   | 8 ++++++++
 bob/bio/face/test/test_transformers.py | 6 ++++++
 3 files changed, 21 insertions(+)

diff --git a/bob/bio/face/test/test_baselines.py b/bob/bio/face/test/test_baselines.py
index 4bff3b8e..b1177216 100644
--- a/bob/bio/face/test/test_baselines.py
+++ b/bob/bio/face/test/test_baselines.py
@@ -12,6 +12,7 @@ import os
 import bob.io.base
 import functools
 import copy
+from nose.plugins.attrib import attr
 
 from bob.bio.base.test.utils import is_library_available
 
@@ -106,31 +107,37 @@ def run_baseline(baseline, samples_for_training=[]):
         assert "scores" in dirs
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_facenet_baseline():
     run_baseline("facenet-sanderberg")
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_inception_resnetv2_msceleb():
     run_baseline("inception-resnetv2-msceleb")
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_inception_resnetv2_casiawebface():
     run_baseline("inception-resnetv2-casiawebface")
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_inception_resnetv1_msceleb():
     run_baseline("inception-resnetv1-msceleb")
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_inception_resnetv1_casiawebface():
     run_baseline("inception-resnetv1-casiawebface")
 
 
+@attr('slow')
 @is_library_available("mxnet")
 def test_arcface_insightface():
     run_baseline("arcface-insightface")
diff --git a/bob/bio/face/test/test_embeddings.py b/bob/bio/face/test/test_embeddings.py
index ca88c315..10b5cf27 100644
--- a/bob/bio/face/test/test_embeddings.py
+++ b/bob/bio/face/test/test_embeddings.py
@@ -4,8 +4,10 @@ import numpy as np
 from bob.pipelines import Sample, wrap
 import pkg_resources
 from bob.bio.base.test.utils import is_library_available
+from nose.plugins.attrib import attr
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_idiap_inceptionv2_msceleb():
     from bob.bio.face.embeddings.tf2_inception_resnet import (
@@ -32,6 +34,7 @@ def test_idiap_inceptionv2_msceleb():
     assert output.size == 128, output.shape
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_idiap_inceptionv2_msceleb_memory_demanding():
     from bob.bio.face.embeddings.tf2_inception_resnet import (
@@ -59,6 +62,7 @@ def test_idiap_inceptionv2_msceleb_memory_demanding():
     assert output.size == 128, output.shape
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_idiap_inceptionv2_casia():
     from bob.bio.face.embeddings.tf2_inception_resnet import (
@@ -85,6 +89,7 @@ def test_idiap_inceptionv2_casia():
     assert output.size == 128, output.shape
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_idiap_inceptionv1_msceleb():
     from bob.bio.face.embeddings.tf2_inception_resnet import (
@@ -111,6 +116,7 @@ def test_idiap_inceptionv1_msceleb():
     assert output.size == 128, output.shape
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_idiap_inceptionv1_casia():
     from bob.bio.face.embeddings.tf2_inception_resnet import (
@@ -137,6 +143,7 @@ def test_idiap_inceptionv1_casia():
     assert output.size == 128, output.shape
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_facenet_sanderberg():
     from bob.bio.face.embeddings.tf2_inception_resnet import (
@@ -163,6 +170,7 @@ def test_facenet_sanderberg():
     assert output.size == 128, output.shape
 
 
+@attr('slow')
 @is_library_available("mxnet")
 def test_arcface_insight_face():
     from bob.bio.face.embeddings.mxnet_models import ArcFaceInsightFace
diff --git a/bob/bio/face/test/test_transformers.py b/bob/bio/face/test/test_transformers.py
index b0506b27..309f344c 100644
--- a/bob/bio/face/test/test_transformers.py
+++ b/bob/bio/face/test/test_transformers.py
@@ -4,6 +4,7 @@ import numpy as np
 from bob.pipelines import Sample, SampleSet
 from bob.bio.base import load_resource
 from bob.bio.base.test.utils import is_library_available
+from nose.plugins.attrib import attr
 
 
 def get_fake_sample(face_size=(160, 160), eyes={"leye": (46, 107), "reye": (46, 53)}):
@@ -13,6 +14,7 @@ def get_fake_sample(face_size=(160, 160), eyes={"leye": (46, 107), "reye": (46,
     return Sample(data, key="1", annotations=annotations)
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_facenet_sanderberg():
     transformer = load_resource("facenet-sanderberg", "transformer")
@@ -24,6 +26,7 @@ def test_facenet_sanderberg():
     assert transformed_sample.data.size == 128
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_inception_resnetv2_msceleb():
     transformer = load_resource("inception-resnetv2-msceleb", "transformer")
@@ -35,6 +38,7 @@ def test_inception_resnetv2_msceleb():
     assert transformed_sample.data.size == 128
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_inception_resnetv2_casiawebface():
     transformer = load_resource("inception-resnetv2-casiawebface", "transformer")
@@ -46,6 +50,7 @@ def test_inception_resnetv2_casiawebface():
     assert transformed_sample.data.size == 128
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_inception_resnetv1_msceleb():
     transformer = load_resource("inception-resnetv1-msceleb", "transformer")
@@ -57,6 +62,7 @@ def test_inception_resnetv1_msceleb():
     assert transformed_sample.data.size == 128
 
 
+@attr('slow')
 @is_library_available("tensorflow")
 def test_inception_resnetv1_casiawebface():
     transformer = load_resource("inception-resnetv1-casiawebface", "transformer")
-- 
GitLab