except PythonLayer from layer factory test
authorEvan Shelhamer <shelhamer@imaginarynumber.net>
Sat, 14 Feb 2015 09:59:43 +0000 (01:59 -0800)
committerJonathan L Long <jonlong@cs.berkeley.edu>
Tue, 17 Feb 2015 06:47:30 +0000 (22:47 -0800)
PythonLayer (rightfully) refuses to be instantiated when a module and
class aren't given. It's still covered by tests however through pytest.

src/caffe/test/test_layer_factory.cpp

index 7a6401c..efb1b37 100644 (file)
@@ -24,6 +24,8 @@ TYPED_TEST(LayerFactoryTest, TestCreateLayer) {
   LayerParameter layer_param;
   for (typename LayerRegistry<Dtype>::CreatorRegistry::iterator iter =
        registry.begin(); iter != registry.end(); ++iter) {
+    // Special case: PythonLayer is checked by pytest
+    if (iter->first == "Python") { continue; }
     layer_param.set_type(iter->first);
     layer = LayerRegistry<Dtype>::CreateLayer(layer_param);
     EXPECT_EQ(iter->first, layer->type());