Merge pull request #3352 from lukeyeager/bvlc/pytest-python-layer

Skip python layer tests if WITH_PYTHON_LAYER unset
This commit is contained in:
Ronghang Hu 2015-11-26 10:45:19 -08:00
Родитель f25f51494a c4190a56ab
Коммит 665649a50d
2 изменённых файлов: 4 добавлений и 0 удалений

Просмотреть файл

@ -77,6 +77,8 @@ def parameter_net_file():
return f.name return f.name
@unittest.skipIf('Python' not in caffe.layer_type_list(),
'Caffe built without Python layer support')
class TestPythonLayer(unittest.TestCase): class TestPythonLayer(unittest.TestCase):
def setUp(self): def setUp(self):
net_file = python_net_file() net_file = python_net_file()

Просмотреть файл

@ -38,6 +38,8 @@ def python_param_net_file():
return f.name return f.name
@unittest.skipIf('Python' not in caffe.layer_type_list(),
'Caffe built without Python layer support')
class TestLayerWithParam(unittest.TestCase): class TestLayerWithParam(unittest.TestCase):
def setUp(self): def setUp(self):
net_file = python_param_net_file() net_file = python_param_net_file()