[tests] Mark tensorflow tests as slow

related to #43

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")

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

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")