Commit c4190a5
committed
Skip python layer tests if WITH_PYTHON_LAYER unset
1 parent 603cbfb commit c4190a5
2 files changed
Lines changed: 4 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -77,6 +77,8 @@ def parameter_net_file(): | |||
| 77 | 77 | return f.name | |
| 78 | 78 | ||
| 79 | 79 | ||
| 80 | + @unittest.skipIf('Python' not in caffe.layer_type_list(), | ||
| 81 | + 'Caffe built without Python layer support') | ||
| 80 | 82 | class TestPythonLayer(unittest.TestCase): | |
| 81 | 83 | def setUp(self): | |
| 82 | 84 | net_file = python_net_file() | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -38,6 +38,8 @@ def python_param_net_file(): | |||
| 38 | 38 | return f.name | |
| 39 | 39 | ||
| 40 | 40 | ||
| 41 | + @unittest.skipIf('Python' not in caffe.layer_type_list(), | ||
| 42 | + 'Caffe built without Python layer support') | ||
| 41 | 43 | class TestLayerWithParam(unittest.TestCase): | |
| 42 | 44 | def setUp(self): | |
| 43 | 45 | net_file = python_param_net_file() | |
0 commit comments